Safety Helmet Wearing Detection Based on YOLOv5 of Attention Mechanism

Abstract

Aiming at problems of low accuracy and strong detection interference of the existing safety helmet wearing detection algorithms, an object detection algorithm by adding the squeeze-and-excitation block based on the YOLOv5 algorithm is proposed in this paper. The proposed method can not only obtain the weight of picture channel, but also accurately separate the foreground and background of the picture. Keeping all parameters unchanged, the proposed method and the YOLOv5 algorithm are applied to detect the safety helmet data set in the experiment. The result shows that the YOLOv5 algorithm with the squeeze-and-excitation block has an average detection accuracy of 94.5% for safety helmets and an average detection accuracy of 92.7% for human heads. The mAP value detected by the proposed method is 2% 2.5% higher than using YOLOv5 algorithm directly.