电子科技大学学报2024,Vol.53Issue(3):414-423,10.DOI:10.12178/1001-0548.2023163
基于改进YOLOv7的矿井人员检测算法
Mine Personnel Detection Algorithm Based on Improved YOLOv7
摘要
Abstract
Real-time detection of mine personnel is an essential part of the construction of intelligent mine.It is of great significance for mine safety production to realize early warning and linkage control of dangerous areas through video monitoring of underground personnel.At present,the visible light image recognition technology needs to be improved for the identification of personnel in the dim environment of underground coal mine.Aiming at the problems of more noise and image blur in the monitoring video caused by uneven illumination and serious coal dust interference in the underground,this paper proposes an improved YOLOv7 mine personnel detection algorithm.Firstly,aiming at the problem of channel isolation caused by direct splicing of ELAN modules,a complex scene detection method based on channel reorganization and feature attention is proposed.Secondly,since the feature fusion results does not focus on the expected target and the model lacks targeted strategies to improve the detection performance of small targets,an ACmix module is added to the neck multi-scale fusion network to take into account both global and local features,which improves the detection ability of the algorithm for small targets.Finally,efficient Intersection over Union(IOU)loss is introduced to improve the convergence speed of the algorithm and reduce the difference between the height and width of the target frame and the prior frame to achieve more accurate positioning.Through the verification of public pedestrian data sets and self-built mine personnel detection data sets,it is shown that the detection accuracy of the proposed algorithm is 3.1%higher than that of the YOLOv7 model,reaching 89.4%;the recall rate is increased by 3.8%to 86.4%,and the speed is increased by 15.8%to 68.8 FPS,meeting the mine personnel real-time detection work requirements.关键词
矿井人员检测/YOLOv7/自注意力/ACmix/损失函数Key words
mine personnel detection/YOLOv7/self-attention/ACmix/loss function分类
矿业与冶金引用本文复制引用
邵小强,李鑫,杨永德,原泽文,杨涛..基于改进YOLOv7的矿井人员检测算法[J].电子科技大学学报,2024,53(3):414-423,10.基金项目
国家自然科学基金(52174198) (52174198)