HWD-YOLO:A New Vision-Based Helmet Wearing Detection Method

It is crucial to ensure workers wear safety helmets when working at a workplace with a high risk of safety accidents,such as construction sites and mine tunnels.Although existing methods can achieve helmet detection in images,their accuracy and speed still need improvements since complex,cluttered,and large-scale scenes of real workplaces cause server occlusion,illumination change,scale variation,and perspective distortion.So,a new safety helmet-wearing detection method based on deep learning is proposed.Firstly,a new multi-scale contextual aggregation module is proposed to aggregate multi-scale feature information globally and highlight the details of concerned objects in the backbone part of the deep neural network.Secondly,a new detection block combining the dilate convolution and attention mechanism is proposed and introduced into the prediction part.This block can effectively extract deep featureswhile retaining information on fine-grained details,such as edges and small objects.Moreover,some newly emerged modules are incorporated into the proposed network to improve safety helmetwearing detection performance further.Extensive experiments on open dataset validate the proposed method.It reaches better performance on helmet-wearing detection and even outperforms the state-of-the-art method.To be more specific,the mAP increases by 3.4%,and the speed increases from17 to 33 fps in comparison with the baseline,You Only Look Once(YOLO)version 5X,and themean average precision increases by 1.0%and the speed increases by 7 fps in comparison with the YOLO version 7.The generalization ability and portability experiment results show that the proposed improvements could serve as a springboard for deep neural network design to improve object detection performance in complex scenarios.

Real-time Safety Helmet-wearing Detection Based on Improved YOLOv5

Safety helmet-wearing detection is an essential part of the intelligentmonitoring system. To improve the speed and accuracy of detection, especiallysmall targets and occluded objects, it presents a novel and efficient detectormodel. The underlying core algorithm of this model adopts the YOLOv5 (YouOnly Look Once version 5) network with the best comprehensive detection performance. It is improved by adding an attention mechanism, a CIoU (CompleteIntersection Over Union) Loss function, and the Mish activation function. First,it applies the attention mechanism in the feature extraction. The network can learnthe weight of each channel independently and enhance the information dissemination between features. Second, it adopts CIoU loss function to achieve accuratebounding box regression. Third, it utilizes Mish activation function to improvedetection accuracy and generalization ability. It builds a safety helmet-wearingdetection data set containing more than 10,000 images collected from the Internetfor preprocessing. On the self-made helmet wearing test data set, the averageaccuracy of the helmet detection of the proposed algorithm is 96.7%, which is1.9% higher than that of the YOLOv5 algorithm. It meets the accuracy requirements of the helmet-wearing detection under construction scenarios.

Small Target HelmetWearing Detection Algorithm Based on Improved YOLO V5

To solve problems such as the low detection accuracy of helmet wear-ing,missing detection and poor real-time performance of embedded equipment in the scene of remote and small targets at the construction site,the text proposes an improved YOLO v5 for small target helmet wearing detection.Based on YOLO v5,the self-attention transformer mechanism and swin transformer module are introduced in the feature fusion step to increase the receptivefield of the con-volution kernel and globally model the high-level semantic feature information extracted from the backbone network to make the model more focused on hel-met feature learning.Replace some convolution operators with lighter and more efficient Involution operators to reduce the number of parameters.The connection mode of the Concat is improved,and 1×1 convolution is added.The experimental results compared with YOLO v5 show that the size of the improved helmet detec-tion model is reduced by 17.8%occupying only 33.2 MB,FPS increased by 5%,and mAP@0.5 reached 94.9%.This approach effectively improves the accuracy of small target helmet wear detection,and meets the deployment requirements for low computational power embedded devices.