2 个月前

基于空间和时间变换器的3D人体姿态估计

Zheng, Ce ; Zhu, Sijie ; Mendieta, Matias ; Yang, Taojiannan ; Chen, Chen ; Ding, Zhengming
基于空间和时间变换器的3D人体姿态估计
摘要

变压器架构已成为自然语言处理领域的首选模型,并且现在正被引入到计算机视觉任务中,如图像分类、目标检测和语义分割。然而,在人体姿态估计领域,卷积架构仍然占据主导地位。在本研究中,我们提出了PoseFormer,这是一种完全基于变压器的方法,用于视频中的3D人体姿态估计,而不涉及卷积架构。受到近期视觉变压器发展的启发,我们设计了一种时空变压器结构,以全面建模每帧内的人体关节关系以及跨帧的时间相关性,从而输出中心帧的精确3D人体姿态。我们对我们的方法在两个流行的标准基准数据集上进行了定量和定性的评估:Human3.6M和MPI-INF-3DHP。大量实验表明,PoseFormer在这两个数据集上均达到了最先进的性能。代码可在以下网址获取:https://github.com/zczcwh/PoseFormer