信阳师范大学学报(自然科学版)2025,Vol.38Issue(2):152-158,7.DOI:10.3969/j.issn.2097-583X.2025.02.005
基于改进YOLOv5s的火焰烟雾检测方法
Flame smoke detection method based on the improved YOLOv5s
摘要
Abstract
To solve the problems of untimely traditional flame smoke detection,difficult pyrotechnic detection and low detection accuracy of small targets,an improved flame smoke detection method of YOLOv5s was proposed.Firstly,the SE attention mechanism was introduced into the backbone layer of the YOLOv5s model,which can adaptively adjust the feature weight of each channel,enhance the useful features and suppresses the useless features,improve the network's ability to extract the features of flame and smoke.Secondly,BiFPN module was introduced as a feature fusion module in the Neck layer of the YOLOv5s model,and the bidirectional connection was introduced through BiFPN module,which can make full use of the feature information of different levels and improve the richness of features by combining the bottom-up and top-down feature fusion methods.Finally,the improved YOLOv5s model was applied to the actual flame smoke dataset,and the experimental results showed that the accuracy,recall rate and mAP value of the improved YOLOv5s model were increased by 1.8%,2.6%and 1.5%,respectively,which can meet the accuracy requirements of flame smoke detection.关键词
火焰烟雾检测/YOLOv5s模型/SE注意力机制/BiFPN模块Key words
flame smoke detection/YOLOv5s model/SE attention mechanism/BiFPN module分类
计算机与自动化引用本文复制引用
孙剑,张数数..基于改进YOLOv5s的火焰烟雾检测方法[J].信阳师范大学学报(自然科学版),2025,38(2):152-158,7.基金项目
国家自然科学青年基金项目(62403405) (62403405)
河南省科技攻关项目(222102210300) (222102210300)