2025年04月06日 星期日 首页   |    期刊介绍   |    编 委 会   |    投稿指南   |    期刊订阅   |    统合信息   |    联系我们
计量学报  2024, Vol. 45 Issue (9): 1314-1323    DOI: 10.3969/j.issn.1000-1158.2024.09.08
  光学计量 本期目录 | 过刊浏览 | 高级检索 |
一种改进YOLOv5s的森林火灾烟雾检测算法
张立国,张琦,金梅,袁煜淋,王泓沣
燕山大学电气工程学院,河北秦皇岛066004
An Improved YOLOv5s Forest Fire Smoke Detection Algorithm
ZHANG Liguo,ZHANG Qi,JIN Mei,YUAN Yulin,WANG Hongfeng
School of Electrical Engineering, Yanshan University, Qinhuangdao, Hebei 066004, China
全文: PDF (843 KB)   HTML (1 KB) 
输出: BibTeX | EndNote (RIS)      
摘要 提出一种基于改进YOLOv5s的森林火灾烟雾检测算法。构建包含16573幅图片的火焰烟雾数据集,解决训练数据不足的问题,提高训练模型的泛化能力。设计一种轻量化的GC-C3模块替换原有的C3模块,减少模型参数量和计算量;将加权双向特征金字塔网络结构融合到Neck结构中,增强网络对于中小目标的检测能力;修改网络空间金字塔池化结构,使用SimSPPF结构替换SPPF,提高了网络的计算效率和检测准确度;将边界框回归损失函数CIOU替换为Focal-EIOU,加快模型的收敛速度,解决正负样本不匹配的问题。实验结果表明:改进之后的网络平均检测准确度提高2.3%,模型参数数量下降46.7%,模型计算量下降47.5%。
服务
把本文推荐给朋友
加入我的书架
加入引用管理器
E-mail Alert
RSS
作者相关文章
张立国
张琦
金梅
袁煜淋
王泓沣
关键词 机器视觉火灾烟雾检测深度学习YOLOv5s轻量化小目标检测Focal-EIOU    
Abstract:A forest fire smoke detection algorithm based on improved YOLOv5s is proposed. A fire and smoke dataset containing 16573 images is constructed to solve the problem of insufficient training data sets and improve the generalization ability of the training model. A lightweight GC-C3 module is designed to replace the original C3 module and reduce the number of model parameters and calculation. The weighted bidirectional feature pyramid network is integrated into the Neck structure to enhance the detection ability of the network for small and medium targets. The network space pyramid pool structure is modified, SPPF is replaced by SimSPPF structure, and the computing efficiency and detection accuracy of the network are improved. The bounding box regression loss function CIOU is replaced by Focal-EIOU to accelerate the convergence of the model and solve the problem of mismatch between positive and negative samples. The experimental results show that the average detection accuracy of the improved network is increased by 2.3%, the number of model parameters is decreased by 46.7%, and the calculation amount of the model is decreased by 47.5%.
Key wordsmachine vision    fire smoke detection    deep learning    YOLOv5s    light weight    small target detection    Focal-EIOU
收稿日期: 2024-02-04      发布日期: 2024-09-26
PACS:  TB96  
  TB973  
基金资助:河北省中央引导地方专项 (199477141G)
作者简介: 张立国(1978-),男,河北秦皇岛人,燕山大学副教授,主要从事计算机视觉、图像处理、智慧工厂研究。Email:zlgtime@163.com
引用本文:   
张立国,张琦,金梅,袁煜淋,王泓沣. 一种改进YOLOv5s的森林火灾烟雾检测算法[J]. 计量学报, 2024, 45(9): 1314-1323.
ZHANG Liguo,ZHANG Qi,JIN Mei,YUAN Yulin,WANG Hongfeng. An Improved YOLOv5s Forest Fire Smoke Detection Algorithm. Acta Metrologica Sinica, 2024, 45(9): 1314-1323.
链接本文:  
http://jlxb.china-csm.org:81/Jwk_jlxb/CN/10.3969/j.issn.1000-1158.2024.09.08     或     http://jlxb.china-csm.org:81/Jwk_jlxb/CN/Y2024/V45/I9/1314
京ICP备:14006989号-1
版权所有 © 《计量学报》编辑部
地址:北三环东路18号(北京1413信箱)  邮编:100029 电话:(010)64271480
本系统由北京玛格泰克科技发展有限公司设计开发  技术支持:support@magtech.com.cn