1. 引言
单目三维人体姿态估计[1]是计算机视觉领域的一项基本任务,旨在从单幅图像或视频中准确检测人体的关键解剖点,以重建完整精确的三维人体姿态模型。这种能力对于包括人类行为分析、人机交互和虚拟/扩增实境在内的广泛应用至关重要[2]-[5]。由于其广泛的实用性,单目三维人体姿态估计一直是学术研究和产业发展的核心焦点。
在大规模三维姿态数据集(如Human3.6M [6]和MPI-INF-3DHP [7])和GPU技术进步的推动下,三维姿态估计的深度学习方法已经取得了重大进展。目前,主流方法[8]通常采用“提升策略”,首先使用先进的二维人体姿态检测算法[9] [10]从图像中提取二维关键点坐标。然后,使用专门设计的模型将这些坐标映射到三维空间,以重建人体三维姿态。然而,仅依赖二维关键点会丢弃固有的图像深度信息,导致深度模糊性和结构不确定性。
为应对上述挑战,研究者转向了端到端的解决方案,其方法演进主要体现在架构创新上:基于时序卷积网络(TCN)的方法[11] [12]更擅长建模局部时序模式,但在捕捉复杂长程依赖上仍面临感受野限制;基于图卷积网络(GCN)的方法[13] [14]虽能利用人体关节的拓扑结构进行空间推理,但其在时序维度上的建模能力通常较为薄弱或依赖额外设计;近期兴起的基于Transformer的方法,凭借其全局自注意力机制,在统一建模长距离时空依赖上展现出显著优势,已成为主流[15] [16]。
然而,标准Transformer的自注意力机制虽然能够根据内容动态加权,但其全局交互模式缺乏对局部时空结构(特别是相邻帧之间强相关性)的显式归纳偏置与约束。这可能导致模型在捕捉运动连续性与姿态平滑性等关键时序模式时效率不足,而这些恰恰是保证姿态序列时域一致性的基础。
为此,本文提出近邻卷积Transformer (NCFormer),其架构如图1所示。其核心是在保留全局建模能力的基础上,创新性地引入轴向卷积来构建一个专注于局部邻域建模的并行机制。具体而言,模型通过三个协同的组件工作:一个多头自注意力模块捕获全局上下文;一个近邻帧卷积模块利用多尺度轴向卷积核聚焦并增强局部帧关系;一个轴向多层感知机对时空特征进行解耦与增强。该设计显式增强了对局部近邻帧的依赖关系建模,从而有效弥补了全局注意力在捕捉精细时序结构上的不足。
Figure 1. NCFormer model architecture diagram
图1. NCFormer模型架构图
2. 相关工作
目前的单目三维人体姿态估计方法主要由两种范式组成:从图像直接对联合坐标进行三维回归[17],以及利用二维姿态检测器[11]-[16]进行二维到三维提升,然后进行三维重建。直接方法通过端到端学习保留空间信息,但通常需要大量的标记数据和计算资源。相比之下,二维到三维方法降低了建模复杂性,增强了稳健性,通常比直接回归达到更高的准确性。然而,这两种范式都面临着持续的挑战,包括二维到三维模糊性、遮挡、深度不确定性、外观变化和复杂的运动动力学。
为应对这些挑战,研究者先后探索了时序卷积网络(TCNs)与图卷积网络(GCNs)。例如,Pavllo等人[12]利用扩张时序卷积捕捉长程依赖;Liu等人[11]引入多尺度时序卷积并结合注意力机制;Cai等人[13]则构建时空图,并通过层次化图卷积提取全局特征。然而,TCN依赖固定卷积核,难以灵活建模多尺度动态模式;GCN通常基于静态图结构,无法适应关节间随时间变化的依赖关系,限制了二者对复杂时空动态的表达能力。
因此,研究焦点逐渐转向基于Transformer的方法。其核心自注意力机制通过全局交互动态建模长程依赖,为统一表征时空信息提供了强大框架。相关研究例如,PoseFormerV2 [8]探索在频域中进行高效且鲁棒的特征表示;P-STMO [15]借助掩码机制进行时空预训练,增强模型对含噪声或遮挡的二维姿态输入的鲁棒性;MHFormer [16]则通过多假设Transformer建模姿态估计中固有的歧义性,以回归出置信度更高的三维姿态。这些方法共同体现了Transformer在处理长序列和挖掘全局上下文方面的卓越能力。
然而,标准Transformer的全局注意力缺乏对近邻帧之间强局部关联的显式归纳偏置,可能导致局部运动连续性被全局上下文稀释,影响序列平滑性与细节捕捉。因此,如何在保持全局建模能力的同时有效加强近邻帧依赖的局部建模,成为关键问题,亦是本文工作的出发点。
为了解决这个问题,我们提出了NCFormer,其架构如图1所示,该方法在继承Transformer全局建模能力的基础上,引入卷积,构建了一个轻量高效的近邻帧卷积模块。该模块通过两次卷积操作,沿时间维度显式地构建近邻帧之间的依赖关系。此外,通过轴向多层感知机对时间、空间与通道维度进行解耦式建模,以抑制跨维度间的特征混杂,从而增强局部表征的判别性与融合效能。
3. 方法
本文提出的三维人体姿态估计方法整体框架如图1所示,图中箭头指示了前向传播的数据流。首先,利用二维关键点检测器从输入视频中提取二维姿态序列。该序列经过嵌入层(Embedding Layer)映射至高维空间,再编码坐标信息(Spatial Position Embedding)。随后,通过空间多头自注意力(Multi-Head Self-Attention)建模关节间的空间依赖,并由轴向多层感知机(Axial Multi-Layer Perceptron)对时间、空间与通道维度进行解耦的独立特征变换。
接下来,将时间信息嵌入特征中,通过时间多头自注意力建立长程时间依赖。之后,近邻帧卷积(Neighboring k Frame Convolution)显式聚合近邻k帧的局部时空信息。在此基础上,轴向多层感知机在时间维度执行独立的解耦变换时:由于特征已融合多帧上下文,该变换能够超越单帧空间结构的局限,在近邻帧信息支撑下更有效地建模帧间连续的局部运动模式,从而实现对时间动态的结构化细化表征。k近邻卷积以及轴向多层感知机的详细结构分别在3.3和3.4节详细讨论。
3.1. 位置编码
标准Transformer中的自注意力机制本质上是置换等价的,不具备对输入序列顺序的感知能力。为使其适用于具有明确时空结构的人体姿态序列,必须显式地向模型中注入关于关节空间位置与帧时序位置的先验知识。
为此,我们参考PoseFomer [18]的思路,分别引入可学习的位置编码。空间位置编码(Spatial Position Embedding)用于区分同一帧内不同的关节点。设人体骨架关节总数为
,我们为每个关节
分配一个可学习的嵌入向量
,嵌入向量集合
。将该嵌入向量嵌入输入,可使编码与不同关节绑定,从而帮助模型识别不同关节的语义。
时间位置编码(Temporal Position Embedding)则用于区分视频序列中的不同时刻。对于长度为
的序列,我们为每一帧
分配一个可学习的嵌入向量
,嵌入向量合集
。将该嵌入向量嵌入输入,从而帮助模型感知动作的时间演进顺序与阶段变化。
3.2. 多头自注意力
多头自注意力机制是Transformer架构的核心思想,对于给定的输入特征
,其中
为批大小,
为帧数,
为关节数,
为特征通道维度。多头注意力的计算首先通过一个线性投影层
(
表示通过线性投影最后维度长度发生的变化)生成查询
(Query)、键
(Key)和值
(Value):
(1)
随后将特征重塑并分割为
个头(head):
(2)
其中
为头数,
为每个头的特征维度。在空间注意力的情况下,注意力在每一帧内部的所有关节之间进行,用于学习关节点之间的空间依赖关系,对于每个头
,注意力权重矩阵通过缩放点积计算:
(3)
在时间注意力的情况下,注意力针对每个独立的关节在所有帧之间进行,用于捕捉关节在时间维度上的运动轨迹。计算前需将
与
维度进行转置,使时间维度成为计算注意力的主体:
(4)
空间注意力权重矩阵,建立了不同关节之间的空间依赖关系,使模型能够学习人体关节之间的相关性,而时间注意力权重矩阵则建立不同时间步之间的全局依赖关系,使模型能够对人体姿态的运动轨迹进行连贯建模。最后加权求和得到多头注意力机制的输出:
(5)
3.3. k近邻卷积
为克服标准Transformer在时间维度缺乏显式局部建模能力的局限,本文提出一个具有强归纳偏置的k近邻时间卷积模块,其结构如图2所示。该模块并非通用卷积,而是专为在时间轴上捕获局部帧间依赖而设计,其核心思想是向模型注入一个关键先验:在时间序列中,相邻帧之间的相关性远高于远距离帧,且此类局部依赖具有连续平滑的结构化特征。
Figure 2. Workflow diagram of k neighboring convolution
图2. k近邻卷积工作流程图
对于给给定输入特征张量
,其中
为批大小,
为通道数,
为帧数(时间维度),
为关节数(空间维度)。在下面的表述中
代表卷积操作。该模块的计算过程可有以下公示链完整刻画:
(6)
(7)
(8)
3.3.1. 时间局部性偏置:时间卷积核设计
我们通过一个一维时间卷积核,显式地注入了时间局部性的强归纳偏置。在实现上,我们使用一个二维卷积核并将其空间宽度约束为1,从而在数学上等价于对每个关节的时序独立进行一维卷积。设输入特征为
,其中
为帧数,
为关节数。我们定义的卷积核参数为
,该设计的关键在于其形状
所编码的硬性约束:k (时间维度):决定了局部时间邻域的宽度,即模型能够观察当前帧前后共k帧的信息;1 (空间维度):强制卷积核在空间维度的感受野为1,这意味着卷积操作不会在同一时间点上跨越不同的关节进行混合。
因此,该卷积操作在物理上等价于对
个关节中的每一个,独立地在其时间序列
上应用一个长度为k的一维卷积。这完美契合了人体运动在短时窗口内连续、平滑的先验知识。
3.3.2. 时间局部性偏置:轴向卷积核设计
该模块采用参数共享的两次卷积结构,注入了对称重建的归纳偏置:首先,通过前向卷积将输入特征从
维投影至
维,并在局部时间窗口内进行信息融合;随后,应用组归一化对通道分组处理,以提升模型对输入噪声的鲁棒性、稳定中间层分布,从而加速收敛;最后,利用同一卷积核的转置进行对称重建,将特征从
维重建为
维。该设计强制前向投影与重建过程共享同一组基,促使模型学习到一组紧凑且具解释性的时间局部特征基。
3.3.3. 在NCFormer中的角色:与全局注意力的偏置协同
在NCFormer架构中,此模块被实例化为近邻帧卷积。它提供了一种与全局多头自注意力互补且正交的归纳偏置,专门负责捕捉帧间运动的微观平滑性与连续性。这种设计使得NCFormer能够同时利用两种强大的模型范式:既拥有Transformer全局建模的灵活性,又具备卷积网络局部建模的效率与结构性。k近邻时间卷积模块确保运动轨迹的局部细节得以精确维护,从而生成更自然、抖动更少的三维姿态序列。
3.4. 轴向多层感知机
为在时空维度上实现解耦的特征变换,同时避免不同维度间的信息干扰,我们设计了一个轴向多层感知机模块,其结构如图3所示。该模块的核心设计思想是:对输入特征的时间、空间和通道维度进行独立的线性变换,从而在保持维度特异性的同时增强特征表示,同时提供了更强的结构化归纳偏置。
Figure 3. Workflow diagram of axial MLP
图3. 轴向多层感知机工作流程图
3.4.1. 维度分离变换机制
维度分离变换机制是轴向MLP与传统MLP的核心区别。该机制通过引入强结构归纳偏置,改变了特征交互的基本范式:传统MLP作为通用函数逼近器,其交互是无结构且无差别的;而轴向MLP通过对时间轴与空间轴进行分离处理,显式注入了时空局部性与维度分离性的先验知识。这使得模型能够分别学习不同维度的特定模式(如时间上的运动平滑性、空间上的关节拓扑关系),避免无关维度信息的干扰,从而显著降低学习难度。
因此,轴向MLP并非简单的多层感知机变体,而是一种专为高维结构化数据(如图像、视频、姿态序列)设计的特征变换层。它将Transformer中“注意力应具备结构性”的思想引入前馈网络,其核心优势在于通过维度解耦,在保持全局感受野的同时,实现更高效、更贴合视觉数据本质的表示学习。
给定输入特征
,其中
为批大小,
为通道数,
为帧数,
为关节数。轴向MLP分别沿两个维度执行独立的线性变换,在下面的公式中
表示线性变换,
表示交换a和b维度:
1) 时间维度变换:
(9)
2) 空间维度变换:
(10)
3.4.2. 特征融合与重构
将空间维度变换
、时间维度变换
、初始输入特征
三个分支的输出沿通道维度拼接后,通过融合层进行降维重构,在下面的公式中,
表示通道维度拼接,
表示线性变换,将最后维度的数据由长度a变化至长度b:
(11)
轴向MLP的设计体现了以下关键优势:轴向MLP的设计具有以下关键优势:首先,其维度解耦偏置通过对时间与空间维度的独立处理,避免了跨维度信息的无差别混合,从而使模型能够更专注地学习各维度特有的模式;其次,三个分支的明确分工赋予了模型良好的结构可解释性——时间分支专注于时序动态,空间分支关注关节间关系,恒等分支则保留原始信息;最后,该模块与注意力机制具备天然的互补性:当与近邻卷积模块协同工作时,轴向MLP的时间分支能够进一步细化由卷积模块提取的局部时间特征,其空间分支则增强空间结构的表示,二者形成“注意力建立依赖,MLP精炼特征”的高效协作模式。
3.5. 损失函数
NCFormer的损失函数
定义为:
(12)
其中
为Zhang等人[19]提出的一种加权平均关节点位置误差,
为Hossain等人[20]提出的一种带有时间平滑性约束的损失函数,
为MPJVE损失,
和
为权重系数,用于调控各损失函数对结果的影响。
的数学公式可以表示为:
(13)
其中
为输入帧序列长度,
为关节数量,
是分配给第
个关节的权重,
是一个权重向量,
和
分别是预测和真实的第
帧第
个关节的三维坐标。其核心思想在于,在计算所有关节的平均位置误差时,为不同关节分配不同的权重,以反映它们在人体姿态估计中的重要程度。比如骨盆、脊柱等根关节的位置误差,会直接导致整个骨架的全局偏移。
和
的数学公式可以表达为:
(14)
(15)
(16)
其中
、
和
分别表示预测的躯干头部、肢体腿部和肢体手臂集合的关节的3D位置,
和
为权重系数,
用于调控预测与真实坐标损失的权重,而
则用于调控预测前后帧坐标差异损失的权重,确保预测序列在时间上平滑。
、
和
则用于调控躯干头部、肢体腿部和肢体手臂集合的权重。
4. 实验
4.1. 数据集以及评估协议
我们在两个具有挑战性的三维人体姿态估计基准数据集Human3.6M [6]和MPI-INF-3DHP [7]上对所提方法进行了全面评估。
Human3.6M是该领域最具代表性和广泛使用的室内数据集之一。为与主流研究保持一致[15] [16],我们使用受试者S1、S5、S6、S7和S8进行训练,并在受试者S9和S11上进行评估。性能评估采用广泛使用的指标,包括平均关节点位置误差(MPJPE)、重建误差(P-MPJPE)以及平均关节点速度误差(MPJVE)。
MPI-INF-3DHP是另一个广泛使用的高质量基准数据集,其特点在于同时涵盖室内和室外场景。为全面评估模型在该数据集上的性能,我们遵循先前工作中建立的标准评估协议[12],采用多种评价指标——包括MPJPE、正确关节点百分比(PCK)和曲线下面积(AUC)。
4.2. 补充细节
我们基于PyTorch框架实现了所提出的NCFormer模型,所有实验均在一张GeForce RTX 4090D GPU上完成。为评估方法的有效性,我们使用由预训练的二维姿态检测器生成的二维关键点[9]或真值二维关键点作为输入。模型使用Adam优化器[21]进行训练,批大小设置为1024,丢弃率为0.1,并采用GELU作为激活函数。初始学习率设为0.00004,衰减因子为0.99。模型共训练了200个epochs。
4.3. Human3.6M数据集评估
Table 1. Performance indicator comparison on Human3.6M Protocol 1 based on 2D key points from the CPN
表1. 基于CPN 2D关键点在Human3.6M协议1上的性能指标对比
模型 |
ECCV20 SRNET [22] |
ICCV21 PoseFormer [18] |
ECCV22 STMO [15] |
CVPR22 MHFormer [16] |
ICCV23 GLA-GCN [23] |
IVC24 STAFormer [24] |
Ours |
方向指示 |
46.6 |
41.5 |
38.9 |
39.2 |
41.3 |
38.0 |
39.5 |
交谈讨论 |
47.1 |
44.8 |
42.7 |
43.1 |
44.3 |
41.8 |
41.7 |
进食 |
43.9 |
39.8 |
40.4 |
40.1 |
40.8 |
39.5 |
36.6 |
打招呼 |
41.6 |
42.5 |
41.1 |
40.9 |
41.8 |
39.8 |
40.0 |
通电话 |
45.8 |
46.5 |
45.6 |
44.9 |
45.9 |
44.8 |
42.8 |
拍照 |
49.6 |
51.6 |
49.7 |
51.2 |
54.1 |
48.9 |
48.3 |
摆姿势 |
46.5 |
42.1 |
40.9 |
40.6 |
42.1 |
39.8 |
41.0 |
购物 |
40.0 |
42.0 |
39.9 |
41.3 |
41.5 |
39.0 |
39.6 |
坐下 |
53.4 |
53.3 |
55.5 |
53.5 |
57.8 |
53.7 |
52.2 |
坐下的过程 |
61.1 |
60.7 |
59.4 |
60.3 |
62.9 |
57.2 |
57.9 |
吸烟 |
46.1 |
45.5 |
44.9 |
43.7 |
45.0 |
43.4 |
42.2 |
等待 |
42.6 |
43.3 |
42.2 |
41.1 |
42.8 |
41.3 |
40.7 |
遛狗 |
43.1 |
46.1 |
42.7 |
43.8 |
45.9 |
41.9 |
40.9 |
行走 |
31.5 |
31.8 |
29.4 |
29.8 |
29.4 |
28.6 |
27.5 |
并肩行走 |
32.6 |
32.2 |
29.4 |
30.6 |
29.9 |
28.7 |
28.3 |
平均 |
44.8 |
44.3 |
42.8 |
43.0 |
44.4 |
41.7 |
41.3 |
我们在Human3.6M数据集上对提出的NCFormer和多种先进的方法进行了全面比较。如表1和表2所示,以CPN检测到的二维关键点作为输入,我们分别使用MPJPE和P-MPJPE指标评估姿态估计精度。表3则进一步通过MPJVE (Mean Per Joint Velocity Error,平均关节速度误差)衡量预测姿态的时间平滑性与一致性。
Table 2. Performance indicator comparison on Human3.6M Protocol 2 based on 2D key points from the CPN
表2. 基于CPN 2D关键点在Human3.6M协议2上的性能指标对比
模型 |
ICCV21 PoseFormer [18] |
ECCV22 P-STMO [15] |
PR23 MHFormer++ [25] |
ICCV23 GLA-GCN [23] |
IVC24 STAFormer [24] |
Ours |
方向指示 |
32.5 |
31.3 |
31.6 |
32.4 |
31.5 |
31.5 |
交谈讨论 |
34.8 |
35.2 |
34.8 |
35.3 |
34.7 |
41.7 |
进食 |
32.6 |
32.9 |
32.2 |
32.6 |
32.5 |
36.6 |
打招呼 |
34.6 |
33.9 |
33.2 |
34.2 |
33.5 |
32.1 |
通电话 |
35.3 |
35.4 |
34.7 |
35.0 |
34.9 |
33.3 |
拍照 |
39.5 |
39.3 |
39.7 |
42.1 |
39.5 |
38.1 |
摆姿势 |
32.1 |
32.5 |
33.0 |
32.1 |
31.8 |
31.5 |
购物 |
32.0 |
31.5 |
31.0 |
31.9 |
31.1 |
30.1 |
坐下 |
32.0 |
44.6 |
43.5 |
45.5 |
43.8 |
42.3 |
坐下的过程 |
48.5 |
48.2 |
49.6 |
49.5 |
47.5 |
46.5 |
吸烟 |
34.8 |
36.3 |
36.1 |
36.1 |
35.8 |
34.3 |
等待 |
32.4 |
32.9 |
32.4 |
32.4 |
32.5 |
31.5 |
遛狗 |
35.3 |
34.4 |
33.8 |
35.6 |
33.9 |
32.5 |
行走 |
24.5 |
23.8 |
23.9 |
23.5 |
23.4 |
22.0 |
并肩行走 |
26.0 |
23.9 |
24.7 |
24.7 |
23.9 |
23.2 |
平均 |
34.6 |
34.4 |
34.2 |
34.8 |
34.0 |
32.9 |
为保障评估的公平性与全面性,我们遵循两种广泛采用的评估协议。在协议1 (基于CPN二维关键点)下,NCFormer取得了最优的MPJPE结果,平均误差为41.3 mm。在协议2 (使用相同CPN关键点)中,我们的方法同样获得最佳性能,平均P-MPJPE为32.9 mm。在两种协议下,NCFormer在绝大多数动作类别上均取得领先或接近最优的表现,这验证了其在全局结构与局部细节建模之间的有效平衡,以及良好的泛化能力。
Table 3. MPJVE performance indicator comparison on Human3.6M based on 2D key points from the CPN
表3. 基于CPN 2D关键点在Human3.6M上的MPJVE性能指标对比
模型 |
ICCV21 PoseFormer [18] |
CVPR22 MHFormer [16] |
CVPR22 MixSTE [19] |
PR23 MHFormer++ [25] |
Ours |
平均 |
2.5 |
2.4 |
2.3 |
2.3 |
2.1 |
在时序一致性方面,NCFormer实现了2.1 mm的MPJVE,显著优其他方法。这一结果表明,我们的方法能够生成具有优越平滑性和时间一致性的运动序列,从动态建模角度体现了其有效捕捉近邻帧依赖的优势。综上,实验说明NCFormer能够基于准确的二维观测恢复出高精度的三维姿态,进一步印证了其模型表达能力与泛化性能。
4.4. MPI-INF-3DHP数据集评估
表4总结了MPI-INF-3DHP数据集的实验结果。我们的方法取得PCK 98.2、AUC 75.7,MPJPE 32.8 mm的优异性能。这些结果证明了我们方法在空间位置预测方面的高准确性,以及在复杂环境和多样化动作类别下强大的泛化能力。值得注意的是,MPI-INF-3DHP数据集涵盖了广泛的室内和室外场景。尽管存在这种变异性,NCFormer始终产生稳定和高质量的预测,突出了其在面对挑战性现实世界条件时的鲁棒性和适应性。上述结果进一步证实了所提方法的有效性与实际应用潜力。
Table 4. Performance comparison on MPI-INF-3DHP database
表4. MPI-INF-3DHP数据集性能对比
模型 |
ICCV21 PoseFormer [18] |
CVPR22 MHFormer [16] |
CVPR22 MixSTE [19] |
PR23 MHFormer++ [25] |
Ours |
PCK↑ |
88.6 |
93.8 |
94.4 |
94.8 |
98.2 |
AUC↑ |
56.4 |
63.3 |
66.5 |
65.8 |
75.7 |
MPJPE↓ (mm) |
77.1 |
58.0 |
54.9 |
54.0 |
32.8 |
4.5. 消融实验
4.5.1. 组件分析
为了评估我们提出模型中每个组件的有效性,我们在Human3.6M数据集上进行了系统的消融实验,并以MPJPE作为评价指标。实验以仅包含位置编码的模型作为基线,逐步添加近邻卷积模块和轴向多层感知机模块,结果如表5所示。
实验表明,当仅使用位置编码作为基线时,MPJPE为47.2 mm。在加入近邻卷积模块后,MPJPE下降至42.3 mm,性能显著提升,说明该模块能有效捕捉近邻帧间的局部依赖关系。进一步引入轴向多层感知机模块后,MPJPE进一步降低至41.3 mm,表明该模块通过对时空维度进行解耦建模,进一步增强了特征的判别力与泛化能力。最终,完整模型(包含所有模块)取得了最优性能。
Table 5. Component ablation experiment
表5. 组件消融实验
模型 |
位置编码 |
k近邻卷积 |
轴向多层感知机 |
MPJPE↓ (mm) |
P-MPJPE↓ (mm) |
MPJVE↓ (mm/s) |
|
√ |
× |
× |
42.5 |
34.0 |
2.53 |
|
√ |
√ |
× |
42.3 |
33.5 |
2.25 |
|
√ |
× |
√ |
41.6 |
33.3 |
2.16 |
Ours |
√ |
√ |
√ |
41.3 |
32.9 |
2.10 |
上述结果验证了近邻卷积模块与轴向多层感知机模块各自的有效性,同时体现了二者在建模时空依赖时的协同作用,共同提升了三维姿态估计的精度与鲁棒性。
4.5.2. 超参数分析
我们在Human3.6M 数据集上对所提方法进行了系统的超参数消融实验,实验结果总结于表6。实验依次考察了模型深度(5、6、7)、嵌入维度(256、512、640)、卷积核大小(5、7、9)以及输入帧长度(81、243、351)对性能的影响。结果表明,深度为6时模型表现最优,优于深度5的较浅结构与深度7的较深结构;嵌入维度设为512时性能最高,优于256与640的设定;卷积核大小为7时效果最佳,优于5和9;而输入帧数为243时预测误差最低,优于81和351帧的配置。
此外,我们分析了各超参数对模型复杂度的影响。其中,嵌入维度与模型深度对参数量和单帧推理计算量(FLOPs)的影响最为显著:嵌入维度因自注意力机制的计算复杂度而带来平方级增长,深度则线性增加模型复杂度。相比之下,卷积核大小(因采用轴向卷积而结构稀疏)与输入帧数对整体复杂度的影响相对较小。
基于上述结果,我们确定了模型深度6、嵌入维度512、卷积核大小7、输入帧数243为最优超参数组合。该配置在所有测试中取得了最低的MPJPE (41.3 mm),表明其能在模型复杂度与表征能力之间实现有效平衡,从而提升姿态估计的准确性。
Table 6. Hyperparameter ablation experiment
表6. 超参数消融实验
深度 |
嵌入维度 |
k |
Frame |
Parameters↓ (M) |
FLOPs↓ (M) |
MPJPE↓ (mm) |
5 |
512 |
7 |
243 |
24.75 |
525.6 |
41.9 |
6 |
512 |
7 |
243 |
29.67 |
630.7 |
41.3 |
7 |
512 |
7 |
243 |
34.59 |
735.8 |
41.5 |
6 |
256 |
7 |
243 |
7.74 |
168.3 |
42.4 |
6 |
640 |
7 |
243 |
46.1 |
972.3 |
41.5 |
6 |
512 |
5 |
243 |
28.1 |
577.3 |
41.8 |
6 |
512 |
9 |
243 |
31.2 |
684.2 |
41.4 |
6 |
512 |
7 |
81 |
29.3 |
605.3 |
42.1 |
6 |
512 |
7 |
351 |
30.1 |
647.7 |
41.5 |
4.6. 定性结果
Figure 4. 3D pose reconstruction comparison (NCFormer vs. PoseFormer) on Human3.6M database
图4. NCFormer与PoseFormer在Human3.6M数据集上的三维姿态重建对比
我们进一步通过定性可视化来验证所提NCFormer模型在三维人体姿态估计上的有效性,从Human3.6M测试集中随机选取若干样本,将NCFormer的三维姿态重建结果与图4中有代表性的Transformer方法PoseFormer结果进行对比。可视化结果清晰表明,在所有动作类别中,NCFormer预测的三维关节位置均更接近真实姿态。相较于其他模型,本方法在肢体长度一致性、关节连接平滑度以及整体结构合理性方面均表现更优。上述可视化结果表明,NCFormer通过有效建模近邻帧依赖关系,显著提升了姿态估计的准确性与鲁棒性。
5. 结论
本文提出了一种新颖的近邻卷积Transformer (NCFormer),用于单目视频中的三维人体姿态估计。NCFormer的核心创新在于其近邻帧依赖捕捉机制:一方面,通过轴向卷积显式地聚合局部近邻帧的时空特征,以捕捉运动连续性与短程依赖;另一方面,通过轴向多层感知机对时间与空间维度进行解耦建模,避免跨维度信息干扰,从而增强特征的结构化表示能力。
实验结果表明,NCFormer在姿态估计精度与运动时序平滑性之间取得了卓越的平衡。在Human3.6M和MPI-INF-3DHP两个权威基准上,NCFormer在MPJPE、P-MPJPE和MPJVE等多个关键指标上达到当前先进方法的性能。定性与定量分析共同验证,我们的方法能够生成更准确、更自然且时间一致性更强的三维姿态序列。
致 谢
我们感谢相关领域研究者公开的代码资源,特别是PoseFormer [18]的工作为我们提供了重要参考。本研究使用的Human3.6M [6]和MPI-INF-3DHP [7]数据集为实验评估提供了基础,在此一并表示感谢。
基金项目
本课题受到“温州大学元宇宙与人工智能研究院”的“重大课题及项目产业化专项资金”(编号:2023103)的资助。
NOTES
*通讯作者。