LAPSE:2024.1987
Published Article

LAPSE:2024.1987
A Lightweight Safety Helmet Detection Algorithm Based on Receptive Field Enhancement
August 28, 2024
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.
Record ID
Keywords
attention mechanism, lightweight, receptive field enhancement module, safety helmet detection, YOLOv5s
Subject
Suggested Citation
Ji C, Hou Z, Dai W. A Lightweight Safety Helmet Detection Algorithm Based on Receptive Field Enhancement. (2024). LAPSE:2024.1987
Author Affiliations
Ji C: School of Electronic and Information Engineering, Liaoning Technical University, Huludao 125105, China [ORCID]
Hou Z: School of Electronic and Information Engineering, Liaoning Technical University, Huludao 125105, China [ORCID]
Dai W: School of Electronic and Information Engineering, Liaoning Technical University, Huludao 125105, China [ORCID]
Hou Z: School of Electronic and Information Engineering, Liaoning Technical University, Huludao 125105, China [ORCID]
Dai W: School of Electronic and Information Engineering, Liaoning Technical University, Huludao 125105, China [ORCID]
Journal Name
Processes
Volume
12
Issue
6
First Page
1136
Year
2024
Publication Date
2024-05-31
ISSN
2227-9717
Version Comments
Original Submission
Other Meta
PII: pr12061136, Publication Type: Journal Article
Record Map
Published Article

LAPSE:2024.1987
This Record
External Link

https://doi.org/10.3390/pr12061136
Publisher Version
Download
Meta
Record Statistics
Record Views
144
Version History
[v1] (Original Submission)
Aug 28, 2024
Verified by curator on
Aug 28, 2024
This Version Number
v1
Citations
Most Recent
This Version
URL Here
https://psecommunity.org/LAPSE:2024.1987
Record Owner
PSE Press
Links to Related Works