何国立, 齐冬莲, 闫云凤. 一种基于关键点检测和注意力机制的违规着装识别算法及其应用[J]. 中国电机工程学报, 2022, 42(5): 1826-1836. DOI: 10.13334/j.0258-8013.pcsee.210282
引用本文: 何国立, 齐冬莲, 闫云凤. 一种基于关键点检测和注意力机制的违规着装识别算法及其应用[J]. 中国电机工程学报, 2022, 42(5): 1826-1836. DOI: 10.13334/j.0258-8013.pcsee.210282
HE Guoli, QI Donglian, YAN Yunfeng. An Illegal Dress Recognition Algorithm Based on Key-point Detection and Attention Mechanism and Its Application[J]. Proceedings of the CSEE, 2022, 42(5): 1826-1836. DOI: 10.13334/j.0258-8013.pcsee.210282
Citation: HE Guoli, QI Donglian, YAN Yunfeng. An Illegal Dress Recognition Algorithm Based on Key-point Detection and Attention Mechanism and Its Application[J]. Proceedings of the CSEE, 2022, 42(5): 1826-1836. DOI: 10.13334/j.0258-8013.pcsee.210282

一种基于关键点检测和注意力机制的违规着装识别算法及其应用

An Illegal Dress Recognition Algorithm Based on Key-point Detection and Attention Mechanism and Its Application

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

     

    Abstract: Existing illegal dress recognition algorithms mainly locate and extract the image features of helmets and work clothes based on the proportion of human body structure, and cannot accurately recognize large postures such as bending, squatting, and climbing. Aiming at the above problems, an illegal dress recognition algorithm based on key-point detection and attention mechanism was proposed, which could be applied to a variety of human postures and different shooting angles, with good performance. Based on pedestrian detection, a single-person key-point detection model was designed, and a region localization strategy was proposed to accurately locate the pedestrian's head, upper body and bottom body areas, which reduced background interference and made dress feature extraction easier. In addition, the attention mechanism was introduced into the basic image classification model to further improve the accuracy of the identification of illegal dress. Experimental results based on human body posture images in substation work scenes verify the effectiveness of the algorithm.

     

/

返回文章
返回