Research on Safety Helmet Detection Algorithm of Power Workers Based on Improved YOLOv5

Abstract

The traditional helmet detection algorithm in power industry has low precision and poor robustness. In response to this problem, the helmet detection algorithm based on improved YOLOv5 (You only look once) is put forward in this paper. Firstly, the YOLOv5 network structure is improved. By increasing the size of the feature map, one scale is added to the original three scales, and the added 160*160 feature map can be used for the detection of small targets; Secondly, the K-means is used for re-clustering the helmet data set to get more suitable priori anchor boxes. The experimental results illustrate that the average accuracy of the improved YOLOv5 algorithm is increased by 2.9% and reaching 95% compared with the initial model, and the accuracy of helmet recognition is increased by 2.4% and reaching 94.6%. This algorithm reduces the rates of missing detection and misdetection of small target detection in original network, and has strong practicability and advanced nature. It can satisfy the requirements of real-time detection and has a certain role in promoting the safety of power industry.