30 May 2023 FM-YOLOv7: an improved detection method for mine personnel helmet
Xiaoqiang Shao, Shibo Liu, Xin Li, Yongde Yang, Zewen Yuan, Hao Li, Zhiyue Lyu
Author Affiliations +
Abstract

Due to the complex underground environment and the small object of the helmet, the detection accuracy is low when the original YOLOv7 algorithm is used to detect whether the mine personnel wears the helmet, which cannot be applied to the actual operation site. In response to this problem, we proposed an FM-YOLOv7 mine personnel helmet detection. First, to improve the feature extraction ability of the shallow network and enhance the representation ability of the model on the helmet, we propose the fused-MBCA (fused-MBConv with the coordinate attention) module. Second, to improve the detection ability of small objects, enable the fused features to obtain high-level semantic information and low-level details from different scales, and have more extensive receptive fields, we propose the multi-scale feature fusion efficient layer aggregation networks. Finally, to accelerate the convergence of the model and improve the regression accuracy, we use efficient intersection over union as the bounding box regression loss function. These experiments are based on the self-built mine personnel safety helmet dataset. The results show that the FM-YOLOv7 model outperformed the other six algorithms. The mAP@0.5 of the proposed model can reach 85.7%, which is 1.4% higher than the original YOLOv7 model. Also, the improved YOLOv7 model achieves 91 frames per second in detection speed, which detects whether the mine personnel wears a safety helmet in real time.

© 2023 SPIE and IS&T
Xiaoqiang Shao, Shibo Liu, Xin Li, Yongde Yang, Zewen Yuan, Hao Li, and Zhiyue Lyu "FM-YOLOv7: an improved detection method for mine personnel helmet," Journal of Electronic Imaging 32(3), 033013 (30 May 2023). https://doi.org/10.1117/1.JEI.32.3.033013
Received: 12 February 2023; Accepted: 15 May 2023; Published: 30 May 2023
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Object detection

Land mines

Feature extraction

Convolution

Detection and tracking algorithms

Safety

Education and training

Back to Top