摘要

现有违规着装识别算法主要根据人体结构比例粗略定位并提取安全帽、工作服的图像特征,存在无法准确识别弯腰、下蹲、攀爬等大姿态问题。针对上述问题,提出一种基于关键点检测和注意力机制的违规着装识别算法,可以适用于多种人体姿态和不同拍摄角度的情况,识别效果好。该算法在行人检测基础上,设计单人关键点检测模型,并且提出区域定位策略,精确定位行人的头部、上半身和下半身区域,减少背景干扰,降低着装特征提取难度。此外,在基础图像分类模型中引入注意力机制,进一步提升违规着装识别的准确率。基于变电站工作场景下人体姿态图像的实验结果验证了该算法的有效性。

全文