基于堆叠沙漏网络改进的多人姿态估计

为解决多人姿态估计中小尺度关键点(手腕、脚踝等)定位准确率低的问题,采用自顶向上的方式,结合先进的人体目标检测模型YOLOv3,提出一种基于堆叠沙漏网络改进的多人姿态估计方法。在沙漏网络中融入通道混洗模块,加强不同尺度下多层特征之间的跨通道信息交流,提高被遮挡关键点的识别效果;使用注意力机制对沙漏网络原有的残差模块进行特征增强,抑制无用特征并提升有用特征,提高小尺度关键点的识别率。实验结果表明,在MPII数据集上的总体PCK@0.5达到了88.6%,在MSCOCO数据集上的AP@0.75相比原始网络提升了4.6%,验证了所提方法的有效性。

Improved multi-person pose estimation based on stacked hourglass networks

To deal with the problem that the low positioning accuracy of small-scale key points(wrist,ankle,etc.)in multi-person pose estimation,a modified multi-person pose estimation algorithm based on stacked hourglass networks was proposed using the top-down method and the advanced human target detection model YOLOv3.The channel shuffling module was integrated into the stacked hourglass networks to enhance the cross-channel information exchange between multi-layer features under diffe-rent scales to improve the recognition effect of occluded key points.Attention mechanism was used to enhance the original residual module of hourglass networks to suppress useless features and enhance useful features,so as to improve the recognition rate of small-scale key points.Experimental results show that the overall PCK@0.5 reaches 88.6%on the MPII datasets.Compared with original networks,the AP@0.75 is improved by 4.6%on the MSCOCO datasets,and the validity of the proposed method is verified.