Helmet wearing detection algorithm based on improved YOLOv5

Yiping Liu,Benchi Jiang,Huan He,Zhijun Chen,Zhenfa Xu

DOI: https://doi.org/10.1038/s41598-024-58800-6

IF: 4.6

2024-04-17

Scientific Reports

Abstract:In industrial production, workers need to wear safety helmets at all times. However, due to different lighting, viewing angles, and the tendency of people to block each other, the precision of target detection is not high enough. Aiming at this problem, a real-time detection of helmets was achieved by improving the YOLOv5 algorithm. This algorithm introduces the lightweight network structure FasterNet, which uses partial convolution as the main operator to reduce the amount of calculations and parameters of the network; the boundary regression loss function Wise-IoU loss function with a dynamic focusing mechanism replaces the original loss function in YOLOv5; finally, the CBAM attention mechanism is introduced to obtain global context information and improve the detection ability of small targets. The experimental results show that the parameters of the improved YOLOv5 model are reduced by 12.68%, the computational amount is reduced by 10.8%, the mAP is increased from 88.3 to 92.3%, and the inference time is reduced by 81.5%, which is better than the performance of the original model and can detect helmet wearing effectively and in real time.

multidisciplinary sciences

What problem does this paper attempt to address?

### Problems the Paper Attempts to Solve

This paper aims to address the issue of insufficient accuracy in detecting workers wearing safety helmets in industrial production. Specifically:

1. **Insufficient Detection Accuracy**: Due to factors such as different lighting conditions, viewing angles, and mutual occlusion of personnel, existing object detection algorithms have low accuracy in detecting the wearing of safety helmets.
2. **Real-time Requirements**: A method capable of real-time detection is needed to meet the safety management needs of industrial sites.
3. **Model Lightweighting**: To deploy on embedded devices, it is necessary to reduce the number of model parameters and computational load to adapt to storage space and power consumption constraints.

To solve the above problems, this paper improves the YOLOv5 algorithm by introducing the lightweight network structure FasterNet, the boundary regression loss function Wise-IoU with a dynamic focusing mechanism, and the CBAM attention mechanism. Experimental results show that the improved YOLOv5 model outperforms the original model in terms of the number of parameters, computational load, mean Average Precision (mAP), and inference time, effectively and in real-time detecting the wearing of safety helmets.