Mine Personnel Detection Algorithm Based on Improved YOLOv7

被引:0
作者
Shao X. [1 ,2 ]
Li X. [1 ,2 ]
Yang Y. [1 ,2 ]
Yuan Z. [1 ,2 ]
Yang T. [1 ,2 ]
机构
[1] College of Electrical and Control Engineering, Xi’an University of Science and Technology, Xi’an
[2] Xi’an Key Laboratory of Electrical Equipment Condition Monitoring and Power Supply Security, Xi’an
来源
Dianzi Keji Daxue Xuebao/Journal of the University of Electronic Science and Technology of China | 2024年 / 53卷 / 03期
关键词
ACmix; loss function; mine personnel detection; self-attention; YOLOv7;
D O I
10.12178/1001-0548.2023163
中图分类号
学科分类号
摘要
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. © 2024 University of Electronic Science and Technology of China. All rights reserved.
引用
收藏
页码:414 / 423
页数:9
相关论文
共 30 条
[1]  
ZHANG K, KANG L, CHEN X, Et al., A review of intelli gent unmanned mining current situation and development trend, Energies, 15, 2, (2022)
[2]  
SHAN P F, LI C W, LAI X P, Et al., Experiment on accurate identification of thermal image of coal-gangue mixture under a simulated dusky and wet condition, Journal of China Coal Society, 48, pp. 802-812, (2023)
[3]  
HE K M, GKIOXARI G, DOLLAR P, Et al., Mask RCNN, Proceedings of the IEEE International Conference on Computer Vision, pp. 2980-2988, (2017)
[4]  
ZHU H G., An efficient lane line detection method based on computer vision, Journal of Physics: Conference Series, 1802, 3, (2021)
[5]  
LI X Y, WANG S, LIU B, Et al., Improved YOLOv4 network using infrared images for personnel detection in coal mines, Journal of Electronic Imaging, 31, 1, (2022)
[6]  
DU Y X, TONG M M, ZHOU L L, Et al., Edge detection based on Retinex theory and wavelet multiscale product for mine images, Applied Optics, 55, 34, pp. 9625-9637, (2016)
[7]  
KOU F R, XIAO W, HE H Y, Et al., Research on tarage detection in underground coal mines based on improved YOLOv5, Journal of Electronics and Information Technology, 45, 7, pp. 2642-2649, (2023)
[8]  
SHAO X Q, LI X, YANG T, Et al., Underground personnel detection and tracking based on improved YOLOv5s and DeepSORT, Coal Science and Technology, 51, 10, pp. 291-301, (2023)
[9]  
LI J Y, ZHAO Y K, XUE Z E, Et al., A survey of model compression for deep neural networks, Chinese Journal of Engineering, 41, 10, pp. 1229-1239, (2019)
[10]  
WANG C Y, BOCHKOVSKIY A, LIAO H Y M., YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors