基于深度学习的安全帽识别算法研究与模型训练

Research and model training of helmet recognition algorithm based on deep learning

针对作业人员不按规定佩戴安全帽和非作业人员误入作业现场的问题,设计了基于深度学习的安全帽和语音识别智能终端算法。对于安全帽的检测,采用了人体关键点检测模型和基于深度学习的YOLO3算法。将智能摄像头得到的视频文件,先利用人体关键点模型提取现场人员图像,再结合YOLO3算法检测现场作业人员佩戴安全帽的情况,对于未正确佩戴安全帽的人员发出告警信息。通过模型训练验证了所提模型的实用性。

There is a problem of workers who don’t wear safety helmets as required as well as non-operating personnel entering a job site by mistake. Thus a deep learning-based safety helmet and voice recognition intelligent terminal algorithm is designed. First, for the detection of helmets, we use a human body key point detection model and a YOLO3 algorithm based on deep learning. The video file obtained by the smart camera is first used to extract the images of the on-site personnel using the human body key point model, and then the YOLO3 algorithm is applied to detect the situation of the on-site workers wearing helmets and send out warning messages for those who do not wear the helmet correctly. Finally, the practicality of the proposed model is verified through model training.