Command Palette
Search for a command to run...
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

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
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| keypoint-detection-on-coco | 4xRSN-50(384×288) | Test AP: 78.6 |
| keypoint-detection-on-coco-test-challenge | 4×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-coco | RSN | AP: 0.792 |
| pose-estimation-on-coco-test-dev | 4xRSN-50 | AP: 78.6 AP50: 94.3 AP75: 86.6 APL: 75.5 APM: 83.3 AR: 83.8 |
| pose-estimation-on-coco-test-dev | 4xRSN-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-pose | 4xRSN-50 | PCKh-0.5: 93.0 |
| pose-estimation-on-mpii-single-person | 4xRSN-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.