融入注意力和密集连接的轻量型人体姿态估计

Lightweight Human Pose Estimation Based on Attention and Dense Connection

目前多数人体姿态估计方法聚焦于提升预测结果的准确性,从而造成了网络参数量大和运算复杂度高等问题。为缓解该矛盾,在高分辨率网络的基础上提出一种融入注意力和密集连接方式的轻量型人体姿态估计网络。重新设计高分辨率网络中的瓶颈模块,从而降低部分网络运算复杂度;改进引入的注意力机制并结合密集连接方式构建了轻量型模块,将其替换高分辨率网络的基础模块,使网络保持一定准确性的同时大幅缩减模型参数量和运算复杂度;利用多分辨率特征和反卷积重新设计网络输出的特征融合方式,最大程度提升模型预测精度。在公开数据集MPII和COCO上的实验结果表明,相比较于高分辨率网络,所提网络模型参数量减少了71.5%,在MPII验证集上,运算复杂度缩小了35.8%,在COCO验证集上,运算复杂度缩小了35.2%,平均准确率提升了0.6个百分点,即网络能在保证检测精度的基础上有效降低网络复杂度。

At present,most human pose estimation methods focus on improving the accuracy of prediction results,which causes problems such as large network parameters and high computational complexity.To alleviate this contradiction,a lightweight human pose estimation network is proposed based on a high-resolution network that integrates attention and dense connections.Firstly,the bottleneck module in the high-resolution network is redesigned to reduce the computational complexity of part of the network.Secondly,the introduced attention mechanism is improved and a light-weight module is constructed in combination with the dense connection method,which replaces the basic module of the high-resolution network so that the network maintains a certain accuracy while greatly reducing the model parameters and computational complexity.Finally,the feature fusion method of the network output is redesigned by using multi-resolution features and deconvolution to maximize the model prediction accuracy.The experimental results on the public datasets MPII and COCO show that,compared with the high-resolution network,the parameters of the proposed network model are reduced by 71.5%.On the MPII validation set,the computational complexity is reduced by 35.8%.On the COCO validation set,the computational complexity is reduced by 35.2%,and the average accuracy is increased by 0.6 percentage points,that is,the network can effectively reduce the network complexity while ensuring detection accuracy.