Affiliation:
1. School of Information and Electromechanical Engineering, Hunan International Economics University, Changsha, China
2. Hunan Institute of Traffic Engineering, Hengyang, China
Abstract
In the construction process, wearing a safety helmet is an important guarantee for personnel safety. However, manual detection is time-consuming, labor-intensive, and unable to provide real-time monitoring. To address this issue, a helmet-wearing detection algorithm has been proposed based on YOLOv5s. The algorithm uses the YOLOv5s network and introduces the CoordAtt coordinate attention mechanism module into its backbone to consider global information and improve the network’s ability to detect small targets. To improve feature fusion, the residual block in the backbone network has been replaced by a Res2NetBlock structure. The experimental results show that compared to the original YOLOv5 algorithm, the accuracy and speed of the self-made helmet data set have improved by 2.3 percentage points and 18 FPS, respectively. Compared to the YOLOv3 algorithm, accuracy and speed have improved by 13.8 percentage points and 95 FPS, respectively, resulting in a more accurate, lightweight, efficient, and real-time helmet-wearing detection.
Subject
Artificial Intelligence,General Engineering,Statistics and Probability