Detection algorithm of safety helmet wearing based on deep learning

• Published in: Concurrency and computation Vol. 33; no. 13
• Main Authors: Huang, Li, Fu, Qiaobo, He, Meiling, Jiang, Du, Hao, Zhiqiang
• Format: Journal Article
• Language: English
• Published: Hoboken, USA John Wiley & Sons, Inc 10-07-2021; Wiley Subscription Services, Inc
• Subjects: Algorithms; Construction site accidents; construction site scene; Deep learning; detection algorithm; Feature maps; Helmets; Image processing; Machine learning; Pixels; safety helmet; Safety helmets; YOLO v3
• ISSN: 1532-0626; 1532-0634
• DOI: 10.1002/cpe.6234

In the production and construction of industry, safety accidents caused by unsafe behaviors of staff often occur. In a complex construction site scene, due to improper operations by personnel, huge safety risks will be buried in the entire production process. The use of deep learning algorithms to replace manual monitoring of site safety regulations is a powerful guarantee for sticking to the line of safety in production. First, the improved YOLO v3 algorithm is used to output the predicted anchor box of the target object, and then pixel feature statistics are performed on the anchor box, and the weight coefficients are respectively multiplied to output the confidence of the standard wearing of the helmet in each predicted anchor box area, according to the empirical threshold determine whether workers meet the standards for wearing helmets. Experimental results show that the helmet wearing detection algorithm based on deep learning in this paper increases the feature map scale, optimizes the prior dimensional algorithm of specific helmet dataset, and improves the loss function, and then combines image processing pixel feature statistics to accurately detect whether the helmet is worn by the standard. The final result is that mAP reaches 93.1% and FPS reaches 55 f/s. In the helmet recognition task, compared to the original YOLO v3 algorithm, mAP is increased by 3.5% and FPS is increased by 3 f/s. It shows that the improved detection algorithm has a better effect on the detection speed and accuracy of the helmet detection task.