HyperAIHyperAI

Command Palette

Search for a command to run...

KTPFormer:基于运动学和轨迹先验知识增强的Transformer用于三维人体姿态估计

Jihua Peng Yanghong Zhou P. Y. Mok

摘要

本文提出了一种新颖的运动学与轨迹先验知识增强的 Transformer 模型(Kinematics and Trajectory Prior Knowledge-Enhanced Transformer,简称KTPFormer),该模型克服了现有基于 Transformer 的三维人体姿态估计方法中Q、K、V向量在自注意力机制中的推导仅依赖于简单线性映射的弱点。我们提出了两种先验注意力模块,即运动学先验注意力模块(Kinematics Prior Attention,简称KPA)和轨迹先验注意力模块(Trajectory Prior Attention,简称TPA),以利用已知的人体解剖结构和运动轨迹信息,促进多头自注意力机制中全局依赖关系和特征的有效学习。KPA通过构建运动学拓扑来建模人体内的运动学关系,而TPA则通过构建轨迹拓扑来学习跨帧的关节运动轨迹信息。这两种模块生成带有先验知识的Q、K、V向量,使得KTPFormer能够同时建模空间和时间相关性。在三个基准数据集(Human3.6M、MPI-INF-3DHP和HumanEva)上进行的大量实验表明,KTPFormer相比现有最先进方法表现出优越的性能。更重要的是,我们的KPA和TPA模块具有轻量级插件设计,可以集成到各种基于 Transformer 的网络(例如扩散型网络)中,在计算开销增加非常小的情况下显著提升性能。代码可在以下地址获取:https://github.com/JihuaPeng/KTPFormer


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供