关键词: 安全帽, 目标检测, 目标跟踪, YOLOv3网络, K-means++聚类, 卡尔曼滤波, KM算法

Abstract: In order to solve the shortcomings of traditional construction site safety management and reduce casualties caused by construction workers not wearing hard hats, a method for detecting and tracking hard hat wearing based on deep learning is proposed. Firstly, the YOLOv3 target detection network is used to realize the helmet wearing detection, and the Kalman filter and the KM algorithm are used to implement multi-target tracking and counting. The test results at a complex construction site show that the detection speed of the network model can reach 45 fps, with an average accuracy of 93%, and the accuracy and recall rates without a helmet are 97% and 95% respectively. This model basically realizes the real-time detection of the wearing condition of the helmet.

Key words: safety helmet, target detection, target tracking, YOLOv3 network, K-means++ clustering, Kalman filtering, KM algorithm