Premium
Detection algorithm of safety helmet wearing based on deep learning
Author(s) -
Huang Li,
Fu Qiaobo,
He Meiling,
Jiang Du,
Hao Zhiqiang
Publication year - 2021
Publication title -
concurrency and computation: practice and experience
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.309
H-Index - 67
eISSN - 1532-0634
pISSN - 1532-0626
DOI - 10.1002/cpe.6234
Subject(s) - algorithm , task (project management) , pixel , feature (linguistics) , computer science , artificial intelligence , object detection , process (computing) , computer vision , pattern recognition (psychology) , engineering , linguistics , philosophy , systems engineering , operating system
Abstract 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.