4 个月前

关键点变换器:解决复杂手部和物体交互中的关节识别问题以实现准确的3D姿态估计

关键点变换器:解决复杂手部和物体交互中的关节识别问题以实现准确的3D姿态估计

摘要

我们提出了一种鲁棒且精确的方法,用于从单张彩色图像中估计两只手在近距离互动时的三维姿态。这是一个极具挑战性的问题,因为可能会出现大面积遮挡以及关节之间的混淆。现有的最先进方法通过为每个关节回归一个热图来解决这一问题,这需要同时解决两个任务:定位关节和识别关节。在这项工作中,我们建议将这些任务分开处理,首先利用卷积神经网络(CNN)将关节定位为二维关键点,然后利用这些关键点处的CNN特征之间的自注意力机制将其与相应的手部关节关联起来。我们称这种架构为“关键点变换器”(Keypoint Transformer),它具有很高的效率,因为在InterHand2.6M数据集上,该模型仅用大约一半的参数数量就达到了最先进水平。此外,我们还展示了该方法可以轻松扩展以高精度估计一只或两只手操纵物体时的三维姿态。为了进一步推动研究,我们创建了一个包含超过75,000张双手操纵物体的图像的新数据集,并对这些图像进行了完整的三维注释,该数据集将公开发布。

代码仓库

shreyashampali/kypt_transformer
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
3d-hand-pose-estimation-on-ho-3dKPT-Transformer
AUC_J: 0.786
PA-MPJPE (mm): 10.8
3d-hand-pose-estimation-on-ho-3d-v3KPT-Transformer
AUC_J: 0.785
PA-MPJPE: 10.9
hand-object-pose-on-ho-3dKeypoint-Trans
ADD-S: 21.4
Average MPJPE (mm): 25.5
OME: 68.0
PA-MPJPE: 10.8
ST-MPJPE: 25.7

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
关键点变换器:解决复杂手部和物体交互中的关节识别问题以实现准确的3D姿态估计 | 论文 | HyperAI超神经