一种基于ODConvBS-YOLOv5s的火焰烟雾检测方法
摘要:
本发明涉及图像处理技术领域,公开了一种基于ODConvBS‑YOLOv5s的火焰烟雾检测方法。包括:获取火焰烟雾图像,并通过labelimgs对图片进行标注制作成数据集,并对其进行预处理;构建改进的YOLOv5s火焰烟雾检测网络,所述YOLOv5s网络包括输入端、骨干网络、Neck模块,所述骨干网络包括CBS模块、CSP1结构和SPPF空间金字塔池化,所述Neck模块采用特征金字塔网络FPN,所述改进的YOLOv5s火焰烟雾检测网络在YOLOv5s网络的骨干网络中前两个CBS模块替换为ODConvBS模块,所述ODConvBS模块包括全维动态卷积ODConv、BN层以及SiLU激活函数;利用改进的YOLOv5s火焰烟雾检测网络进行火焰烟雾检。与现有技术相比,本发明检测精度高、检测速度快、漏检率低、收敛速度快。
公开/授权文献
0/0