基于YOLOv4的安全帽佩戴检测及工种身份识别

作者：王晨^*; 齐华; 史建利

来源：计算机系统应用, 2022, 31(07): 272-277.

DOI：10.15888/j.cnki.csa.008561

• 安全帽佩戴检测
• YOLOv4
• K-means算法
• 多尺度检测

在施工现场中,安全帽能够减轻对头部的伤害,且不同颜色的安全帽代表不同的身份,基于当前施工现场通过视频监控来对工人安全帽的佩戴以及工种身份识别存在一定的耗时性,不完全性,监督效率低等问题,本文提出了一种基于YOLOv4改进的安全帽佩戴检测以及身份识别的方法,在原始的YOLOv4的基础之上,使用K-means算法对先验框的大小重新进行聚类分析处理,增加多尺度预测输出,实现DIoU NMS进行非极大值抑制,从而使工人安全帽佩戴及身份识别达到高效性,全面性.结果表明,佩戴红、蓝、黄、白安全帽和未佩戴安全帽工人平均检测准确率达到92.1%,从而保证能够实现对施工现场工人安全帽的佩戴达到一种实时监控.

• 单位
  西安工业大学; 电子信息工程学院