Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Detection, Pose Estimation and Segmentation for Multiple Bodies: Closing the Virtuous Circle

About

Human pose estimation methods work well on isolated people but struggle with multiple-bodies-in-proximity scenarios. Previous work has addressed this problem by conditioning pose estimation by detected bounding boxes or keypoints, but overlooked instance masks. We propose to iteratively enforce mutual consistency of bounding boxes, instance masks, and poses. The introduced BBox-Mask-Pose (BMP) method uses three specialized models that improve each other's output in a closed loop. All models are adapted for mutual conditioning, which improves robustness in multi-body scenes. MaskPose, a new mask-conditioned pose estimation model, is the best among top-down approaches on OCHuman. BBox-Mask-Pose pushes SOTA on OCHuman dataset in all three tasks - detection, instance segmentation, and pose estimation. It also achieves SOTA performance on COCO pose estimation. The method is especially good in scenes with large instances overlap, where it improves detection by 39% over the baseline detector. With small specialized models and faster runtime, BMP is an effective alternative to large human-centered foundational models. Code and models are available on https://MiraPurkrabek.github.io/BBox-Mask-Pose.

Miroslav Purkrabek, Jiri Matas• 2024

Related benchmarks

TaskDatasetResultRank
Pose EstimationCOCO (val)
AP76.8
319
Pose EstimationOCHuman (test)
AP79.9
95
Instance SegmentationOCHuman (test)
Mask AP34
38
Instance SegmentationOCHuman (val)
Mask AP33.7
25
Pose EstimationOCHuman (val)
AP79
24
Object DetectionOCHuman (val)
mAP49
17
Object DetectionOCHuman (test)
mAP48.8
17
DetectionOCHuman (test)
bbox AP35.9
7
Object DetectionCIHP (val)
AP69.7
7
Instance SegmentationCIHP (val)
AP65.9
6
Showing 10 of 13 rows

Other info

Code

Follow for update