基于人体关键点的安全带违规佩戴检测

摘要

为对施工环境中脚手架工作人员安全带佩戴的规范性做出检测,提出了一种安全带违规佩戴检测算法。算法由目标检测和语义分割两个部分组成。在目标检测部分,利用CenterNet网络作为主体架构,由不同的预测分支完成人体关键点以及脚手架旋转目标定位任务。在语义分割部分,为了提升识别精度,在原始Deep Lab V3+网络上增加了特征融合结构以及改进后的CBAM-f注意力模块实现了挂绳的像素级定位任务,改进后的网络相对于原始模型在m IOU和F1分数上分别提升了3.5%和2.7%。