一种鲁棒的轻量级人体姿态估计模型

A ROBUST LIGHTWEIGHT HUMAN POSE ESTIMATION MODEL

  • 摘要: 遮挡是影响人体估计鲁棒性的重要因素。在Lightweight OpenPose模型基础上,提出一种鲁棒的轻量级人体姿势估计模型,利用多分辨率表征融合模块来增强对局部特征的提取;通过GCABlock获取长距离上下文信息判断被遮挡关键点的坐标;同时利用姿态调整机对全局和局部信息进行平衡以达到最大贡献。通过对初始和细化阶段进行轻量化改进,减少参数量并提升定位的精度。训练时使用BoneLoss来增加网络对人体约束信息的识别。实验结果表明,提出的GMNet可以有效检测出被遮挡位置的姿态,相较于OpenPose模型参数减少至近5.5百分点,检测速度提升约6倍,精度约为原模型的92.6%。

     

    Abstract: Occlusion is an important factor affecting the robustness of human estimation. This paper proposes a robust lightweight human pose estimation model based on Lightweight OpenPose. It utilized multi-resolution representation fusion module to enhance extraction of local features. It obtained long-distance context information through GCABlock to judge coordinates of occluded key points, and simultaneously used attitude adjustment machine to balance global and local information to achieve maximum contribution. By making lightweight improvements to initial and refinement stages, the number of parameters was reduced and positioning accuracy was improved. BoneLoss was used during training to increase network recognition of human body constraints information. Experimental results show that GMNet can effectively detect occluded poses, with parameters reduced to nearly 5.5% of OpenPose model and detection speed improved by about 6 times, and accuracy reaches about 92.6% of original model.

     

/

返回文章
返回