基于结构重参数化的人体姿态估计算法

HUMAN POSE ESTIMATION ALGORITHM BASED ON STRUCTURAL REPARAMETERIZATION

  • 摘要: 当前基于深度学习的高精度人体姿态估计算法训练及推理效率低下,针对此问题,设计轻量级人体姿态估计模型RepHRNet。提出基于结构重参数化的Rep-shuffleblock模块,提升模型推理及训练速度。提出Lite fuse layer特征融合层,提升多尺度特征图融合效率。实验证明,该模型相较于轻量级高分辨率网络LiteHRNet,在MPII数据集及COCO数据集上实现了更快的训练推理速度及更高的精度。

     

    Abstract: The current high-precision human pose estimation algorithm is inefficient in training and inference. To solve this problem, a lightweight human pose estimation model RepHRNet is designed. A Rep-shuffle block module based on structural reparameterization was proposed to improve the inference and training speed. The Lite fuse layer feature fusion layer was proposed to improve the efficiency of feature map fusion. Experiments show that compared with the lightweight high-resolution network LiteHRNet, the model achieves faster training and inference speed and higher accuracy on MPII dataset and COCO dataset.

     

/

返回文章
返回