从多人多视角视频一次性连贯重建人体与场景 Coherent Human-Scene Reconstruction from Multi-Person Multi-View Video in a Single Pass
无需外部模块,一次性从多人多视角视频联合重建相机、场景和人体
前置知识
SMPL模型
SMPL(Skinned Multi-Person Linear)是一个参数化的人体身体模型,它使用姿态参数θ和体型参数β来控制人体形状和姿态。SMPL-X是SMPL的扩展版本,增加了手部和面部关节,能够表示更精细的全身动作。模型将人体表示为三角形网格,通过混合蒙皮技术将网格顶点与骨架关联,实现自然的姿态变化。
本文使用SMPL-X作为人体表示,理解其参数结构对于理解人体重建模块的工作原理至关重要,特别是作者如何分离视图不变参数和视图相关参数进行融合。
Structure-from-Motion (SfM)
SfM是一种从多张图像中恢复相机姿态和三维点云的经典方法。它通过特征点匹配和束调整来优化相机内参、外参和三维点位置。传统SfM基于几何优化,计算成本高且在动态场景中表现不佳。
本文与SfM解决相同的问题,但采用数据驱动的前馈神经网络替代迭代优化,实现8倍以上的速度提升。理解SfM有助于理解本文的创新之处。
Multi-View Stereo (MVS)
MVS是SfM的后续步骤,利用已知的相机姿态从多视角图像重建密集的三维表面。它通过匹配像素对应关系和深度估计来生成高精度的三维几何。传统MVS方法需要复杂的优化过程。
本文的Pi3X模型可以看作是MVS的现代替代方案,使用神经网络直接预测点云。理解MVS有助于理解本文在场景重建方面的贡献和对比基线的选择。
Head-Pelvis长度
头骨中心到骨盆中心的三维距离是人体尺度的稳定参考值,因为它对姿态变化相对不变。在图像空间中,这对应于检测到的头部关键点和骨盆关键点之间的像素距离。在SMPL模型中,这对应于投影后的头部关节和骨盆关节之间的距离。
这是本文尺度调整模块的核心思想。Pi3X预测的场景是近似度量尺度的,而SMPL是度量尺度的,两者存在尺度不匹配。作者通过计算图像中头骨-骨盆长度与SMPL投影长度的比率来调整场景尺度,确保人体与场景的一致整合。
Sim(3)变换
Sim(3)表示相似变换群,包含旋转、平移和均匀缩放。在3D重建评估中,使用Sim(3)对齐可以消除全局尺度和位姿差异,专注于重建的相对准确性。相比之下,SE(3)只包含旋转和平移,不包含缩放。
本文使用WA-MPJPE和W-MPJPE等指标,它们分别使用Sim(3)和SE(3)对齐。理解这些对齐方式对于正确解读实验结果至关重要。
射线三角测量
射线三角测量是从多视角图像中确定三维点位置的经典方法。给定相机参数和图像中的2D观测,每个视角定义一条从相机中心穿过观测点的射线,这些射线的交点即为三维点的位置。在存在噪声的情况下,使用最小二乘法求解最优交点。
本文在多视图融合中不直接平均预测的3D头部位置,而是使用射线三角测量计算全局位置,这强制了多视角一致性并提高了精度。理解这一技术对于理解本文的视图融合策略至关重要。
研究动机
现有的人体场景重建方法存在几个关键问题。首先,大多数方法如JOSH3R、UniSH和Human3R仅支持单目输入,无法利用多视角提供的丰富几何信息。其次,尝试扩展到多视角的方法如HSfM和HAMSt3R需要额外的开销模块(如2D关键点检测器、跨视角重识别模块)或预处理数据(如跨视角人员身份),这增加了系统复杂性和计算成本。例如,HSfM需要使用人体关节作为对应关系来校准相机姿态,HAMSt3R需要预测分割、密集姿态和场景几何,但这些方法都依赖迭代优化过程。此外,基于外观的重识别方法在视觉相似的场景(如穿着制服的人群)中往往不可靠,导致错误的跨视角关联。
本文的目标是本文的目标是提出CHROMM,一个统一的框架,能够从多人多视角视频联合重建相机参数、场景点云和人体网格,且无需依赖外部模块或预处理数据。具体而言,模型应该直接从RGB输入视频输出相机空间点地图、相机参数以及共享所有视角的全局SMPL-X参数,包括姿态参数、形状参数、全局根旋转和全局3D头部位置。
与已有工作不同的是,本文的独特切入角度在于不将人体重建和场景重建视为两个独立的子问题,而是将Pi3X(3D基础模型)和Multi-HMR(人体网格恢复)集成到一个统一的可训练神经网络架构中。与需要迭代优化的方法不同,本文采用前馈架构在单次传递中完成所有重建。与依赖外观关联的方法不同,本文提出基于几何的多人员关联方法,利用3D位置和人体姿态等显式几何线索建立跨视角的身份对应关系。此外,针对Pi3X预测近似度量尺度场景而SMPL是度量尺度导致的尺度不匹配问题,本文创新性地提出使用头骨-骨盆长度比率进行尺度调整,而非简单的全局尺度参数。
核心方法
CHROMM的整体思路是采用双编码器架构,分别优化场景几何和人体细节。对于每个输入帧,Pi3X编码器提取场景特征,Multi-HMR编码器提取人体特征。场景特征通过Pi3X解码器处理,由相机头部和点部分别预测相机参数和局部3D点地图。人体特征绕过Pi3X解码器,直接传递到人体重建头部。作者从人体特征中检测包含人类头部的补丁,收集对应的人体令牌,并从解码后的场景特征图中采样相应的令牌,通过MLP融合产生人体令牌。每个令牌通过SMPL解码器回归SMPL参数。在测试时,模型执行多视图融合和多人员关联,将单视图估计聚合为一致的全球表示,然后使用尺度调整模块对齐人体和场景。
CHROMM的核心创新点在于其统一的前馈架构和测试时优化策略。与Human3R等早期融合方法不同,作者有意避免在场景令牌和人体令牌之间进行早期融合,因为这会改变解码器的输入分布,即使使用冻结权重也会负面影响几何重建性能。另一个关键创新是视图不变和视图依赖组件的分离处理:形状参数和规范空间中的姿态参数是视图不变的,直接平均预测;根旋转和3D头部位置是视图依赖的,首先转换到共享的世界坐标系,然后使用射线三角测量计算全局位置。第三个创新是基于几何的多人员关联,使用3D位置和规范空间姿态的加权组合作为匹配成本,而不是依赖外观特征。最后,针对Pi3X和SMPL之间的尺度不匹配,作者提出使用头骨-骨盆长度比率进行尺度调整,这比直接使用Pi3X预测的全局尺度因子更加稳定。
方法步骤详情
CHROMM的训练和推理分为几个步骤。在编码阶段,给定多视角视频帧,模型首先将它们展平为单个序列。对于每个输入图像,双编码器分别提取场景特征和人体特征。场景特征被分割为补丁令牌,通过Pi3X解码器的交替注意力层处理,而人体特征绕过解码器。在解码阶段,解码后的场景令牌输入到相机头部和点部分别回归相机参数和局部3D点地图。同时,从人体特征中检测头部补丁,收集相应的人体令牌,从解码后的场景特征图中采样相应令牌,通过MLP融合产生人体令牌。每个人体令牌通过SMPL解码器回归SMPL参数,通过预测2D头部关键点和深度残差来恢复3D头部位置。在测试时,首先执行多人员关联,使用匈牙利算法和基于3D位置和姿态的成本函数匹配跨视角的个体。然后执行多视图融合,对视图不变参数直接平均,对视图依赖参数转换到世界坐标系后使用射线三角测量计算全局位置。最后,使用尺度调整模块,计算图像中头骨-骨盆长度和SMPL投影长度,通过比率调整全局尺度。
技术新颖性
CHROMM的技术新颖性体现在多个方面。首先,据作者所知,这是第一个统一框架,能够在单次传递中联合重建相机、场景和多人,无需外部模块或预处理数据。其次,尺度调整模块使用头骨-骨盆长度比率,比直接使用Pi3X预测的度量标记更稳定地处理Pi3X和SMPL之间的尺度差距。第三,测试时多视图融合策略将人类表示分解为视图不变和视图依赖组件,分别处理,避免了混合视图依赖特征的token级池化。第四,基于几何的多人员关联方法使用3D位置和规范空间姿态的加权组合(位置权重0.8,姿态权重0.2),比外观基方法在视觉相似场景中更鲁棒。最后,从深度估计重构头部位置,使用点部分预测的深度地图作为先验,预测深度残差而非绝对深度,这稳定了训练并提高了泛化能力。
实验结果
CHROMM在多个数据集上展示了显著的性能提升和效率优势。在全局人体运动估计任务上,使用EMDB-2数据集的单目设置中,CHROMM-mono达到WA-MPJPE 102.6mm、W-MPJPE 255.0mm、RTE 1.7%,优于JOSH3R(WA-MPJPE 220.0mm、W-MPJPE 661.7mm、RTE 13.1%)、UniSH(WA-MPJPE 118.5mm、W-MPJPE 270.1mm、RTE 5.8%)和Human3R(WA-MPJPE 112.2mm、W-MPJPE 267.9mm、RTE 2.2%)。在RICH数据集的多视角设置中,CHROMM-multi达到WA-MPJPE 53.1mm、W-MPJPE 79.0mm、RTE 1.4%,相比单目设置有显著提升(WA-MPJPE 87.5mm、W-MPJPE 138.3mm、RTE 3.3%)。在多视角人体姿态估计任务上,使用EgoHumans数据集,CHROMM达到W-MPJPE 0.51m、GA-MPJPE 0.15m、PA-MPJPE 0.05m,优于HSfM(W-MPJPE 1.04m、GA-MPJPE 0.21m、PA-MPJPE 0.05m)和HAMSt3R(W-MPJPE 3.80m、GA-MPJPE 0.42m、PA-MPJPE 0.14m)。使用EgoExo4D数据集,CHROMM达到W-MPJPE 0.26m、PA-MPJPE 0.06m,同样优于HSfM(W-MPJPE 0.56m、PA-MPJPE 0.06m)和HAMSt3R(W-MPJPE 0.51m、PA-MPJPE 0.09m)。在运行时比较中,CHROMM在单个NVIDIA V100 GPU上处理3个人、4个视角的单个时间步需要约4秒,相比HSfM(约118秒)和HAMSt3R(约32秒)实现超过8倍的加速。消融研究表明,尺度调整模块将EMDB-2上的WA-MPJPE从169.7mm降低到102.6mm,W-MPJPE从447.9mm降低到255.0mm,RTE从4.2%降低到1.7%。多视图融合策略中,平均预测和三角测量相结合达到最佳性能(WA-MPJPE 53.1mm、W-MPJPE 79.0mm、RTE 1.4%)。多人员关联策略中,结合位置和姿态线索达到最高准确率91.3%。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Global Human Motion Estimation (EMDB-2) | WA-MPJPE ↓ | 102.6mm | Human3R: 112.2mm | 8.5% |
| Global Human Motion Estimation (EMDB-2) | RTE(%) ↓ | 1.7% | Human3R: 2.2% | 22.7% |
| Global Human Motion Estimation (RICH multi-view) | WA-MPJPE ↓ | 53.1mm | Human3R (mono): 110.0mm | 51.7% |
| Multi-View Human Pose Estimation (EgoHumans) | W-MPJPE ↓ | 0.51m | HSfM: 1.04m | 51.0% |
| Multi-View Human Pose Estimation (EgoExo4D) | W-MPJPE ↓ | 0.26m | HAMSt3R: 0.51m | 49.0% |
| Runtime (EgoHumans single frame) | Total Runtime ↓ | 4s | HAMSt3R: 32s | 8× faster |
局限与改进
作者在结论中承认CHROMM的一个共同限制:与Multi-HMR和Human3R类似,它严重依赖头部令牌。因此,当头部区域被严重遮挡或不可见时,性能可能会下降。作者计划将双编码器扩展为集成编码器设计,以同时表示人体和场景。除了作者承认的限制外,另一个潜在限制是尺度调整模块假设头骨-骨盆长度在图像中可检测。当人体被严重裁剪(如只显示上半身或下半身)时,骨盆可能位于图像边界之外,作者在这种情况下使用从头部令牌估计的粗略骨盆预测,但这可能引入额外误差。此外,模型在多人员关联中需要构建视角连通图,这可能在高密度场景中变得复杂。最后,模型使用两阶段训练策略,第一阶段冻结Pi3X主干和Multi-HMR编码器,这可能限制了整体优化空间。
独立分析的弱点
CHROMM在几个特定场景中存在弱点。首先,当头部被严重遮挡或多人重叠时,头部令牌检测可能失败,导致人体重建和关联错误。一个改进方向是增加身体其他关键部位(如手部、脚部)作为补充检测目标,提供冗余信息。其次,尺度调整模块在人体被严重裁剪时可能不准确,因为骨盆检测依赖于可见的头部令牌。改进方向可以是使用更多尺度参考点(如肩膀到臀部距离)或学习一个鲁棒的尺度预测网络。第三,模型在高密度多人员场景中可能出现身份交换,因为几何匹配在相似姿态和位置的个体之间可能模糊。改进方向可以是结合几何线索和轻量级外观特征,提高区分能力。第四,两阶段训练策略限制了整体优化,特别是Pi3X和Multi-HMR编码器的适应能力。改进方向可以是端到端微调,使用渐进式解冻策略。最后,模型不处理时间一致性优化,虽然通过跟踪实现了一定的时序连贯性,但在快速运动或遮挡切换时可能出现跳跃。改进方向可以是引入时序平滑约束或光流引导的匹配。
未来方向
作者提出的未来方向是将双编码器扩展为集成编码器设计,以同时表示人体和场景。基于本文成果,可以延伸几个研究方向。首先,探索动态场景重建,目前模型只重建静态场景点云,未来可以扩展到重建动态物体(如移动的车辆、动物)。其次,研究实时推理优化,虽然模型已经比优化基线快8倍,但在实时应用中仍然需要进一步加速。第三,扩展到更多传感器模态,如结合深度传感器或事件相机,提高重建精度和鲁棒性。第四,研究大规模训练策略,目前使用BEDLAM和组合的野外数据集,未来可以探索合成数据和自监督学习的结合,减少对标注数据的依赖。第五,研究可解释性,特别是尺度调整模块和关联策略的决策过程,这对于关键应用(如自动驾驶、医疗)很重要。最后,探索下游应用,如机器人导航、AR/VR内容生成和体育分析,验证模型的实用价值。
复现评估
作者提到项目页面(https://nstar1125.github.io/chromm)提供了补充材料,但摘要中未明确说明代码是否开源。关于数据集,模型使用BEDLAM进行第一阶段训练,结合3DPW、MPII和MSCOCO构建野外训练集进行第二阶段训练,这些数据集都是公开可用的。评估使用EMDB、RICH、EgoHumans和EgoExo4D,也都是公开数据集。关于算力,运行时比较在单个NVIDIA V100 GPU上进行,处理3个人、4个视角的单个时间步需要约4秒。训练过程使用两阶段策略,第一阶段冻结Pi3X主干和Multi-HMR编码器,训练SMPL解码器、融合、掩码和骨盆检测MLP;第二阶段解冻骨盆检测MLP,保持其他模块固定。这种模块化训练策略有助于控制训练复杂度和稳定性。总体而言,复现难度中等偏高,主要挑战在于获取和预处理多视角数据集以及配置双编码器架构。如果作者开源代码和预训练模型,复现难度将显著降低。
论文图表
这张图展示了CHROMM的整体概念。左侧显示输入的多视角视频,右侧显示重建的相机、场景点云和人体网格。图中间用箭头表示从输入到输出的单次传递处理,直观地展示了CHROMM的核心能力:无需分阶段处理,直接从多人多视角视频同时重建相机参数、三维场景和多个精确的人体网格模型。
这张图对理解论文至关重要,因为它简洁地概括了CHROMM的核心贡献:统一的、单次传递的人体场景重建框架。与需要多阶段处理、外部模块或迭代优化的现有方法形成鲜明对比,这张图直观地展示了本文的独特价值主张。通过展示输入和输出的直接对应关系,读者可以立即理解模型的目标和能力。