nav emailalert searchbtn searchbox tablepage yinyongbenwen piczone journalimg journalInfo journalinfonormal searchdiv searchzone qikanlogo popupnotification paper paperNew
2024, 06, v.54 1454-1461
基于改进YOLOv5的城市火灾检测算法研究
基金项目(Foundation): 国家自然科学基金(U19A2061); 吉林省科技厅中青年科技创新创业卓越人才(团队)项目(创新类)(20220508133RC); 吉林省科技发展计划项目(20210404020NC)~~
邮箱(Email):
DOI:
摘要:

火灾引发的经济与人员损伤始终是社会的棘手问题,迫切需要能够实时、准确监控火灾发生的方案。针对城市火灾场景复杂、目标小和定位要求高等问题,提出了一种改进YOLOv5的城市火灾场景下烟火目标检测算法。整理收集到的网络数据、构建数据集,并进行数据增强。基于YOLOv5s算法模型,重构网络结构,增加小目标检测层,使模型更加关注小目标的检测。嵌入了压缩与激励网络(Squeeze-and-Excitation Network, SENet),使YOLOv5模型的检测精度进一步提升。讨论了SENet添加位置的问题。实验结果表明,改进YOLOv5算法的精确率达到了93.7%,与原YOLOv5s相比召回率和平均精确度分别提高了1.9%、1.6%;在网络中添加注意力模块的位置不同,所产生的模型效果也不同。

Abstract:

The economic losses and personnel injuries caused by fires have always been a thorny issue in society, and there is an urgent need for solutions that can monitor the occurrence of fires in real-time and accurately. A modified YOLOv5 algorithm for detecting pyrotechnic targets in urban fire scenes is proposed to address the problems of complexity, small targets, and high positioning requirements in urban fire scenes. Firstly, the collected network data is organized, the dataset is constructed, and the data enhancement is performed. Then, based on the YOLOv5s algorithm model, the network structure is reconstructed and a small object detection layer is added to make the model pay more attention to small object detection. Finally, the Squeeze-and-Excitation Network(SENet) are embedded to further improve the detection accuracy of the YOLOv5 model. In addition, the issue of adding locations for SENet is also discussed. The experimental results show that the accuracy of the improved YOLOv5 algorithm has reached 93.7%, and compared with the original YOLOv5s, the recall rate and average accuracy have increased by 1.9% and 1.6%; it is found that adding the attention module at different locations in the network produces different modeling effects.

参考文献

[1] 应急管理部举行习近平总书记向国家综合性消防救援队伍授旗致训词4周年专题新闻发布会[Z/OL].(2022-10-31)[2023-07-10].https://www.mem.gov.cn/xw/xwfbh/2022n10y31rxwfbh/wzsl_4260/202210/t202

21031_425195.shtml.

[2] 刘颖,刘红燕,范九伦,等.基于深度学习的小目标检测研究与应用综述[J].电子学报,2020,48(3):590-601.

[3] LIU W,ANGUELOV D,ERHAN D,et al.SSD:Single Shot MultiBox Detector[C]//Computer Vision-ECCV 2016.Amsterdam:Springer,2016:21-37.

[4] REDMON J,DIVVALA S,GIRSHICK R,et al.You Only Look Once:Unified,Real-time Object Detection[EB/OL].(2015-06-08)[2023-07-11].https://arxiv.org/abs/1506.02640.

[5] REDMON J,FARHADI A.YOLOv3:An Incremental Improvement[EB/OL].(2018-04-08)[2023-07-11].https://arxiv.org/abs/1804.02767.

[6] GIRSHICK R.Fast R-CNN[C]//2015 IEEE International Conference on Computer Vision (ICCV).Santiago:IEEE,2015:1440-1448.

[7] REN S Q,HE K M,GIRSHICK R,et al.Faster R-CNN:Towards Real-time Object Detection with Region Proposal Networks[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2017,39(6):1137-1149.

[8] 宁阳,杜建超,韩硕,等.改进DeeplabV3+的火焰分割与火情分析方法[J].西安电子科技大学学报,2021,48(5):38-46.

[9] 常丽,张雪,蒋辉,等.融合YOLOv5s与SRGAN的实时隧道火灾检测[J].电子测量与仪器学报,2022,36(8):223-230.

[10] WOO S,PARK J,LEE J Y,et al.CBAM:Convolutional Block Attention Module[C]//15th European Conference on Computer Vision.Munich:Springer,2018:3-19.

[11] 喻丽春,刘金清.基于改进Mask R-CNN的火焰图像识别算法[J].计算机工程与应用,2020,56(21):194-198.

[12] 章曙光,唐锐,邵政瑞,等.一种基于小波变换的YOLOv5火灾检测改进算法[J].无线电工程,2023,53(10):2303-2310.

[13] 刘洪江,王懋,刘丽华,等.基于深度学习的小目标检测综述[J].计算机工程与科学,2021,43(8):1429-1442.

[14] 牛为华,殷苗苗.基于改进YOLO v5的道路小目标检测算法[J].传感技术学报,2023,36(1):36-44.

[15] 韩俊,袁小平,王准,等.基于YOLOv5s的无人机密集小目标检测算法[J].浙江大学学报(工学版),2023,57(6):1224-1233.

[16] 刘展威,陈慈发,董方敏.基于YOLOv5s的航拍小目标检测改进算法研究[J].无线电工程,2023,53(10):2286-2294.

[17] 贾志,李茂军,李婉婷.基于改进YOLOv5+DeepSort算法模型的交叉路口车辆实时检测[J].计算机工程与科学,2023,45(4):674-682.

[18] 汪颖,王峰,李玮,等.用于复杂环境下果蔬检测的改进YOLOv5算法研究[J].中国农机化学报,2023,44(1):185-191.

[19] 杨国亮,杨浩,余帅英,等.改进YOLOv5的交通标志检测算法[J].计算机工程与应用,2023,59(10):262-269.

[20] LIN T Y,DOLLáR P,GIRSHICK R,et al.Feature Pyramid Networks for Object Detection[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition.Honolulu:IEEE,2017:936-944.

[21] HU J,SHEN L,ALBANIE S,et al.Squeeze-and-Excitation Networks[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2020,42(8):2011-2023.

基本信息:

中图分类号:TP391.41;X932

引用信息:

[1]武慧,杨玉竹,卜显峰,等.基于改进YOLOv5的城市火灾检测算法研究[J].无线电工程,2024,54(06):1454-1461.

基金信息:

国家自然科学基金(U19A2061); 吉林省科技厅中青年科技创新创业卓越人才(团队)项目(创新类)(20220508133RC); 吉林省科技发展计划项目(20210404020NC)~~

检 索 高级检索

引用

GB/T 7714-2015 格式引文
MLA格式引文
APA格式引文