Safety Helmet Detection Using Improved YOLOv8
The wearing of safety helmet directly affects the life safety of workers in the construction site or production workshop, and it is of great significance for the supervision of workers’ helmet wearing. In this paper, for the current helmet wearing detection algorithm in the complex environment detection accuracy is not high, put forward a yolov8-based improved helmet detection algorithm YOLOv8-Focaler-SE, in the network structure to add SE attention mechanism module, so that the network more accurately locate the region of interest in the target, at the same time in the CIOU loss function on the basis of the introduction of the Focaler-CIoU loss, thus thereby thereby improving the overall performance of the algorithm. Experimental results on the safety helmet wearing dataset show that YOLOv8-Focaler-SE improves the precision, recall, mAP 50 and mAP $50-95$ metrics by $1.2 \%$, $2.1 \%, 1.5 \%$, and $2.1 \%$, respectively, while the number of model parameters remains essentially unchanged when compared to the yolov8n algorithm. The method improves the performance of the detection algorithm, which can accurately detect helmet wearing and provide technical support for safety production.
Authors
- Shaodong Sun (ORCID: https://orcid.org/0009-0005-1284-3421)
- Bin Yao (ORCID: https://orcid.org/0000-0003-0818-7044)
- Zhao Chen
- Baoping Liu
- Xiang Gao
Institutions
- State Grid Corporation of China (China) (CN)
Publication Details
- Published
- 2024-09-20
- DOI
- https://doi.org/10.1109/icbase63199.2024.10762189
- Primary Topic
- Fire Detection and Safety Systems
- Type
- article
- Field-Weighted Citation Impact
- 0.00