首页|多场景烟雾环境下改进的YOLOv5s烟雾检测算法

多场景烟雾环境下改进的YOLOv5s烟雾检测算法

Improved YOLOv5s smoke detection algorithm in multi-scenario smoke environment

扫码查看
烟雾检测往往检测精度低、漏检率与误检率高,为了解决这些问题,提出一种改进的YOLOv5 s的烟雾检测模型.首先将C3_PSA结构添加到YOLOv5 s的骨干部分中,提高模型在恶劣的环境下对烟雾边界的特征提取能力,降低漏检率;其次,将BiFusion的结构替换YOLOv5 s模型的颈部结构,增强模型对细节特征以及位置信息的检测能力,提高模型检测精度,降低模型误检率;最后,对损失函数进行改进以进一步提高模型检测准确率.实验结果显示,改进后的模型的精度提升了5.6%,平均精度均值(mAP)提升了 3.5%,FPS 为 369 帧/秒,说明改进后的模型可以在背景环境复杂时准确地检测出烟雾特征,同时满足模型在端侧部署的实时性高的要求.
Smoke detection often has low detection accuracy,high missed detection rate and false de-tection rate,in order to solve these problems,an improved YOLOv5s smoke detection model was pro-posed.Firstly,the C3_PSA structure was added to the backbone of YOLOv5s to improve the feature ex-traction ability of the model in harsh environments and reduce the missed detection rate.Secondly,BiFu-sion 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.

deep learningimage processingPSAsmoke detection

骈璐璐、裴焕斗、张宇璇

展开 >

中北大学 仪器与电子学院,山西 太原 030051

百信信息技术有限公司,山西 太原 030006

深度学习 图像处理 PSA 烟雾检测

2024

工业仪表与自动化装置
陕西鼓风机(集团)有限公司

工业仪表与自动化装置

CSTPCD
影响因子:0.393
ISSN:1000-0682
年,卷(期):2024.(2)
  • 11