A Lightweight Safety Helmet Detection Algorithm Based on Receptive Field Enhancement

Changpeng Ji,Zhibo Hou,Wei Dai

DOI: https://doi.org/10.3390/pr12061136

IF: 3.5

2024-05-31

Processes

Abstract:Wearing safety helmets is an important way to ensure the safety of workers' lives. To address the challenges associated with low accuracy, large parameter values, and slow detection speed of existing safety helmet detection algorithms, we propose a receptive field-enhanced lightweight safety helmet detection algorithm called YOLOv5s-CR. First, we use a lightweight backbone, a high-resolution feature fusion network, and a small object detection layer to improve the detection accuracy of small objects while substantially decreasing the model parameters. Next, we embed a coordinate attention mechanism into the feature extraction network to improve the localization accuracy of the detected object. Finally, we propose a new receptive field enhancement module (RFEM) to substitute the SPPF module in the original network, enabling the model to acquire features under multiple receptive fields, thereby enhancing the detection precision of multi-scale objects. Using the Safety Helmet Detection dataset for validation, in contrast to the initial YOLOv5s, the parameters of the improved algorithm were reduced by 62.8% to 2.61 M, and P, R, and mAP0.5 were increased by 1.5%, 1.2%, and 2.0%, respectively. The detection speed can reach 149FPS on the RTX3070 GPU, which satisfies the accuracy and real-time requirements for detecting safety helmets.

engineering, chemical

What problem does this paper attempt to address?

This paper focuses on the issue of safety helmet detection, particularly as an important measure to ensure the safety of workers in industrial production. Existing safety helmet detection algorithms face challenges such as low accuracy, large parameter size, and slow detection speed. To address these challenges, the authors propose a lightweight safety helmet detection algorithm based on receiver field enhancement - YOLOv5s-CR.

First, they use a lightweight backbone network, high-resolution feature fusion network, and small object detection layer to improve the detection accuracy of small objects while significantly reducing the model parameters. Second, they embed a coordinate attention mechanism into the feature extraction network to improve the localization accuracy of the detection objects. Finally, they design a novel receiver field enhancement module (RFEM) to replace the original SPPF module in the network, enabling the model to capture features in multiple receiver fields and improve the detection accuracy of multi-scale objects.

The improved algorithm is validated on the Safety Helmet Detection dataset. Compared with the original YOLOv5s, the parameters are reduced by 62.8% to 2.61 M. The precision (P), recall (R), and mean average precision (mAP) at 0.5 IoU threshold are improved by 1.5%, 1.2%, and 2.0% respectively. The detection speed reaches 149 FPS, meeting the real-time requirements while maintaining accuracy.

The paper also discusses the limitations of existing safety helmet detection algorithms, such as large model parameters and slow detection speed. Through experiments and comparisons, the effectiveness of the proposed algorithm is demonstrated, showing improved detection performance and efficiency. This has important implications for industrial safety management and the application of deep learning technology in this field.