Safety helmet detection method based on YOLO v4

In the actual production process, safety accidents caused by workers not wearing safety helmets often occur. In order to reduce safety accidents caused by wearing helmets, a helmet detection method based on improved YOLO v4 is proposed. By collecting a self-made data set of on-site construction site...

Full description

Saved in:
Bibliographic Details
Published in2020 16th International Conference on Computational Intelligence and Security (CIS) pp. 155 - 158
Main Authors Benyang, Deng, Xiaochun, Lei, Miao, Ye
Format Conference Proceeding
LanguageEnglish
Published IEEE 01.11.2020
Subjects
Adaptation models
Head
Image edge detection
neural network
Production
Real-time systems
Safety
safety helmet detection
target detection
Training
YOLO v4
Online AccessGet full text

Cover

More Information
Summary:In the actual production process, safety accidents caused by workers not wearing safety helmets often occur. In order to reduce safety accidents caused by wearing helmets, a helmet detection method based on improved YOLO v4 is proposed. By collecting a self-made data set of on-site construction site video, using the K-means algorithm to cluster the data set in getting appropriately a priori frame dimensional center and obtaining more targeted edge information. Subsequently, a multi-scale training strategy is used in the network training process to improve the adaptability of the model from different scales of detection. The experimental results show that, in the helmet wearing detection task, the model mAP value reached 92.89%, the detection speed reached 15f/s, and its detection accuracy and detection speed were improved compared with YOLO v4, which satisfies the real-time requirements of the helmet detection task.
DOI:10.1109/CIS52066.2020.00041