无线电工程2024，Vol.54Issue(6)：1454-1461,8.DOI:10.3969/j.issn.1003-3106.2024.06.014
基于改进YOLOv5的城市火灾检测算法研究
Research on Urban Fire Detection Algorithm Based on Improved YOLOv5
摘要
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.关键词
火灾检测/小目标样本/YOLOv5/压缩与激励网络Key words
fire detection/small target samples/YOLOv5/SENet分类
信息技术与安全科学引用本文复制引用
武慧,杨玉竹,卜显峰,曹丽英..基于改进YOLOv5的城市火灾检测算法研究[J].无线电工程,2024,54(6):1454-1461,8.基金项目
国家自然科学基金(U19A2061) （U19A2061）
吉林省科技厅中青年科技创新创业卓越人才(团队)项目(创新类)(20220508133RC) （团队）
吉林省科技发展计划项目(20210404020NC)National Natural Science Foundation of China(U19A2061) （20210404020NC）
Young and Middle-aged Scientific and Technological Innovation and Entre-preneurship Excellence Talent(Team)Project(Innovation Category)of Jilin Provincial Science and Technology Department(20220508133RC) （Team）
Project of Jilin Provincial Science and Technology Development Plan(20210404020NC) （20210404020NC）
