Paper
21 July 2023 Helmet wearing detection based on YOLOv5
Author Affiliations +
Proceedings Volume 12717, 3rd International Conference on Artificial Intelligence, Automation, and High-Performance Computing (AIAHPC 2023); 127170F (2023) https://doi.org/10.1117/12.2684745
Event: 3rd International Conference on Artificial Intelligence, Automation, and High-Performance Computing (AIAHPC 2023), 2023, Wuhan, China
Abstract
Complex environments, such as dense personnel and background interference, affect the detection accuracy of whether personnel wear helmets. To solve this problem, a new detection algorithm for helmets based on YOLOv5 is studied in this paper. Firstly, RepVGGblock is used to replace the common convolution in the network to effectively utilize the computing power of GPU. Secondly, the efficient channel attention mechanism is incorporated into the C3 module of the backbone network to enhance the feature identification ability of the backbone network for the hard hat. Then, the boundary box regression function is changed to SIoU to redefine the distance loss and improve the regression accuracy. Finally, the detection performance of multi-scale targets is improved by adding detection layers. The test results on the self-made safety helmet data set show that the average accuracy of the improved YOLOv5 model reaches 94.6%, which is 4.1% higher than that of the original model, and can meet the requirements of target detection.
© (2023) COPYRIGHT Society of Photo-Optical Instrumentation Engineers (SPIE). Downloading of the abstract is permitted for personal use only.
Chao Wu, De-yong Wang, Wen-xi Shi, Jian Fang, Xue-yi Zhao, and Yan-yun Fu "Helmet wearing detection based on YOLOv5", Proc. SPIE 12717, 3rd International Conference on Artificial Intelligence, Automation, and High-Performance Computing (AIAHPC 2023), 127170F (21 July 2023); https://doi.org/10.1117/12.2684745
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Object detection

Convolution

Detection and tracking algorithms

Target detection

Deep learning

Small targets

Back to Top