Keypoint Detection
sapiens
English
rawalkhirodkar's picture
Update README.md
119c31d verified
metadata
language: en
license: cc-by-nc-4.0
pipeline_tag: keypoint-detection
tags:
  - sapiens

Pose-Sapiens-1B-Bfloat16

Model Details

Sapiens is a family of vision transformers pretrained on 300 million human images at 1024 x 1024 image resolution. The pretrained models, when finetuned for human-centric vision tasks, generalize to in-the-wild conditions. Sapiens-1B natively support 1K high-resolution inference. The resulting models exhibit remarkable generalization to in-the-wild data, even when labeled data is scarce or entirely synthetic.

  • Developed by: Meta
  • Model type: Vision Transformer
  • License: Creative Commons Attribution-NonCommercial 4.0
  • Task: pose
  • Format: bfloat16
  • File: sapiens_1b_goliath_best_goliath_AP_639_bfloat16.pt2

Model Card

  • Image Size: 1024 x 768 (H x W)
  • Num Parameters: 1.169 B
  • FLOPs: 4.647 TFLOPs
  • Patch Size: 16 x 16
  • Embedding Dimensions: 1536
  • Num Layers: 40
  • Num Heads: 24
  • Feedforward Channels: 6144

More Resources

Uses

Pose 1B model can be used for estimate 308 keypoints (body + face + hands + feet) on a single image.