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

Sapiens: Foundation for Human Vision Models

About

We present Sapiens, a family of models for four fundamental human-centric vision tasks -- 2D pose estimation, body-part segmentation, depth estimation, and surface normal prediction. Our models natively support 1K high-resolution inference and are extremely easy to adapt for individual tasks by simply fine-tuning models pretrained on over 300 million in-the-wild human images. We observe that, given the same computational budget, self-supervised pretraining on a curated dataset of human images significantly boosts the performance for a diverse set of human-centric tasks. The resulting models exhibit remarkable generalization to in-the-wild data, even when labeled data is scarce or entirely synthetic. Our simple model design also brings scalability -- model performance across tasks improves as we scale the number of parameters from 0.3 to 2 billion. Sapiens consistently surpasses existing baselines across various human-centric benchmarks. We achieve significant improvements over the prior state-of-the-art on Humans-5K (pose) by 7.6 mAP, Humans-2K (part-seg) by 17.1 mIoU, Hi4D (depth) by 22.4% relative RMSE, and THuman2 (normal) by 53.5% relative angular error. Project page: https://about.meta.com/realitylabs/codecavatars/sapiens.

Rawal Khirodkar, Timur Bagautdinov, Julieta Martinez, Su Zhaoen, Austin James, Peter Selednik, Stuart Anderson, Shunsuke Saito• 2024

Related benchmarks

TaskDatasetResultRank
Pose EstimationCOCO (val)
AP66.1
319
Pose EstimationOCHuman (test)
AP70.5
95
Pose EstimationOCHuman (val)
AP69.8
24
Surface Normal EstimationHi4D
MAE12.14
18
Depth EstimationTHuman Face 2.1 (test)
RMSE0.0112
15
Depth EstimationTHuman UpperBody 2.1 (test)
RMSE0.0145
15
Depth EstimationTHuman FullBody 2.1 (test)
RMSE0.0172
15
Depth EstimationHi4D (test)
RMSE0.106
15
Pose EstimationHumans-5K (test)
Body AP64.7
13
Video Depth EstimationHi4D
OPW0.0122
13
Showing 10 of 20 rows

Other info

Code

Follow for update