4 个月前

多上下文注意力机制用于人体姿态估计

多上下文注意力机制用于人体姿态估计

摘要

在本文中,我们提出了一种将卷积神经网络与多上下文注意力机制结合的方法,用于端到端的人体姿态估计框架。我们采用了堆叠的沙漏网络来从多个分辨率和不同语义的特征中生成注意力图。条件随机场(CRF)被用来建模注意力图中相邻区域之间的相关性。此外,我们将整体注意力模型(关注整个人体的全局一致性)和身体部位注意力模型(关注不同身体部位的详细描述)结合起来。因此,我们的模型具备了从局部显著区域到全局语义一致空间的不同粒度的关注能力。另外,我们设计了新颖的沙漏残差单元(Hourglass Residual Units, HRUs),以扩大网络的感受野。这些单元是残差单元的扩展,通过侧支引入具有更大感受野的滤波器,从而在HRUs内部学习并结合不同尺度的特征。我们在两个广泛使用的人体姿态估计基准数据集上评估了所提出的多上下文注意力机制和沙漏残差单元的有效性。实验结果表明,我们的方法在所有身体部位上均优于现有的所有方法。

代码仓库

bearpaw/pose-attention
pytorch
GitHub 中提及

基准测试

基准方法指标
pose-estimation-on-leeds-sports-posesMulti-Context Attention
PCK: 92.6%
pose-estimation-on-mpii-human-poseMulti-Context Attention
PCKh-0.5: 91.5

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
多上下文注意力机制用于人体姿态估计 | 论文 | HyperAI超神经