摘要
烟雾检测往往检测精度低、漏检率与误检率高,为了解决这些问题,提出一种改进的YOLOv5s的烟雾检测模型。首先将C3_PSA结构添加到YOLOv5s的骨干部分中,提高模型在恶劣的环境下对烟雾边界的特征提取能力,降低漏检率;其次,将BiFusion的结构替换YOLOv5s模型的颈部结构,增强模型对细节特征以及位置信息的检测能力,提高模型检测精度,降低模型误检率;最后,对损失函数进行改进以进一步提高模型检测准确率。实验结果显示,改进后的模型的精度提升了5.6%,平均精度均值(mAP)提升了3.5%,FPS为369帧/秒,说明改进后的模型可以在背景环境复杂时准确地检测出烟雾特征,同时满足模型在端侧部署的实时性高的要求。
Smoke detection often has low detection accuracy,high missed detection rate and false detection rate,in order to solve these problems,an improved YOLOv5s smoke detection model was proposed.Firstly,the C3_PSA structure was added to the backbone of YOLOv5s to improve the feature extraction ability of the model in harsh environments and reduce the missed detection rate.Secondly,BiFusion structure was used to replace the neck structure of YOLOv5s model,so as to enhance the model′s ability to detect details and location information,improve the model detection accuracy,and reduce the model false detection rate.Finally,the loss function is improved to further improve the accuracy of model detection.The experimental results show that the accuracy of the improved model is improved by 5.6%,the mAP is improved by 3.5%,and the FPS is 369 frames/second,which indicates that the improved model can accurately detect the smoke characteristics in the complex background environment,and at the same time meet the requirements of high real-time deployment of the model on the end side.
作者
骈璐璐
裴焕斗
张宇璇
PIAN Lulu;PEI Huandou;ZHANG Yuxuan(School of Instrument and Electronics,North University of China,Shanxi Taiyuan 030051,China;100 Trust Information Technology Co.,LTD.,Shanxi Taiyuan 030006,China)
出处
《工业仪表与自动化装置》
2024年第2期101-105,共5页
Industrial Instrumentation & Automation