基于改进YOLO-V5算法的烟火检测方法  

Firework detection method based on improved YOLO-V5 algorithm

作　　者：张明振 段江忠 梁肇伟 郭俊杰 柴大山 ZHANG Mingzhen;DUAN Jiangzhong;LIANG Zhaowei;GUO Junjie;CHAI Dashan(Shenzhen Urban Public Safety and Technology Institute,Shenzhen Guangdong 518038,China;School of Big Data and Internet,Shenzhen Technology University,Shenzhen Guangdong 518118,China;Shenzhen Branch of China Tower,Shenzhen Guangdong 518000,China)

机构地区：[1]深圳市城市公共安全技术研究院有限公司,广东深圳518038 [2]深圳技术大学大数据与互联网学院,广东深圳518118 [3]中国铁塔股份有限公司深圳市分公司,广东深圳518000

出　　处：《中国安全科学学报》2024年第5期155-161,共7页China Safety Science Journal

摘　　要：为减少自然环境中云、水雾、沙尘、灯光、日出、日落等干扰因素对烟雾、火焰目标检测准确性的影响,提出一种基于改进YOLO-V5算法的烟火检测算法。采用现场采集和网络爬取的方法获取烟雾、火焰目标图像和干扰类图像数据集,均衡学习训练样本,提高模型泛化能力;使用加权双向特征金字塔网络(BiFPN)替换原有的特征金字塔网络(FPN)+路径聚合网络(PAN)结构,对目标进行多尺度特征融合,加强模型特征融合能力;同时,运用距离交并比(DIoU)非极大值抑制(NMS)替代原有的NMS,加快检测框损失函数收敛速度,加强模型推理能力。结果表明:改进后的算法准确率为79.2%,召回率为68.6%,平均精度均值(mAP)为74.2%,误报率(FPR)为12.8%;相比于原YOLO-V5算法,改进后的算法准确率、召回率、mAP分别提高1.9%、0.9%、2.7%,检测识别FPR降低3.7%。To reduce the influences of background interference factors in natural environments such as clouds,mist,dust,lights,sunrise,and sunset on the smoke and flame target detection accuracy,a smoke and fire detection algorithm based on an improved YOLO-V5 algorithm was proposed.Smoke,flame target images,and interference image data sets were obtained from the on-site collection and web crawling approaches to solve sample imbalance and improve model generalization ability.A bidirectional feature pyramid network(BiFPN)was used to replace the original feature pyramid network(FPN)+path aggregation network(PAN)structure,and then multi-scale feature fusion on the target was performed to strengthen the model feature fusion ability.At the same time,distance intersection-over-union(DIoU)non-maximum suppression(NMS)is used to replace the original non-maximum suppression(NMS)to speed up the convergence of the detection box loss function and enhance the model reasoning ability.The results showed that the improved algorithm's accuracy,recall rate,mean average precision(mAP)and FPR were 79.2%,68.6%,74.2%,and 12.8%,respectively.Compared with the original YOLO-V5 algorithm,the proposed algorithm improved accuracy rate,recall rate,and mAP by 1.9%,0.9%,and 2.7%,respectively.Furthermore,the FPR was decreased by 3.7%.

关 键 词：YOLO-V5算法 烟雾 火焰 目标检测 误报率(FPR) 

分 类 号：X932[环境科学与工程—安全科学]