HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Learning Delicate Local Representations for Multi-Person Pose Estimation

Yuanhao Cai Zhicheng Wang Zhengxiong Luo Binyi Yin Angang Du Haoqian Wang Xiangyu Zhang Xinyu Zhou Erjin Zhou Jian Sun

Learning Delicate Local Representations for Multi-Person Pose Estimation

Abstract

In this paper, we propose a novel method called Residual Steps Network (RSN). RSN aggregates features with the same spatial size (Intra-level features) efficiently to obtain delicate local representations, which retain rich low-level spatial information and result in precise keypoint localization. Additionally, we observe the output features contribute differently to final performance. To tackle this problem, we propose an efficient attention mechanism - Pose Refine Machine (PRM) to make a trade-off between local and global representations in output features and further refine the keypoint locations. Our approach won the 1st place of COCO Keypoint Challenge 2019 and achieves state-of-the-art results on both COCO and MPII benchmarks, without using extra training data and pretrained model. Our single model achieves 78.6 on COCO test-dev, 93.0 on MPII test dataset. Ensembled models achieve 79.2 on COCO test-dev, 77.1 on COCO test-challenge dataset. The source code is publicly available for further research at https://github.com/caiyuanhao1998/RSN/

Code Repositories

chenyilun95/tf-cpn
tf
Mentioned in GitHub
caiyuanhao1998/RSN
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
keypoint-detection-on-coco4xRSN-50(384×288)
Test AP: 78.6
keypoint-detection-on-coco-test-challenge4×RSN-50
AP: 77.1
AP50: 93.3
AP75: 83.6
APL: 82.6
AR: 82.6
AR50: 96.1
AR75: 88.2
ARL: 88.7
ARM: 78.0
multi-person-pose-estimation-on-cocoRSN
AP: 0.792
pose-estimation-on-coco-test-dev4xRSN-50
AP: 78.6
AP50: 94.3
AP75: 86.6
APL: 75.5
APM: 83.3
AR: 83.8
pose-estimation-on-coco-test-dev4xRSN-50 (ensemble)
AP: 79.2
AP50: 94.4
AP75: 87.1
APL: 76.1
APM: 83.8
AR: 84.1
pose-estimation-on-mpii-human-pose4xRSN-50
PCKh-0.5: 93.0
pose-estimation-on-mpii-single-person4xRSN-50
PCKh@0.5: 93

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp