Image Segmentation
sapiens
English
rawalkhirodkar's picture
Update model card for Sapiens with architecture details
f446dbf verified
|
raw
history blame
1.42 kB
---
language: en
license: cc-by-nc-4.0
---
# Seg-Sapiens-0.3B-Torchscript
### 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-0.3B 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:** seg
- **Format:** torchscript
- **File:** sapiens_0.3b_goliath_best_goliath_mIoU_7673_epoch_194_torchscript.pt2
### Model Card
- **Image Size:** 1024 x 768 (H x W)
- **Num Parameters:** 0.336 B
- **FLOPs:** 1.242 TFLOPs
- **Patch Size:** 16 x 16
- **Embedding Dimensions:** 1024
- **Num Layers:** 24
- **Num Heads:** 16
- **Feedforward Channels:** 4096
### More Sources
- **Repository:** [https://github.com/facebookresearch/sapiens](https://github.com/facebookresearch/sapiens)
- **Paper:** [https://arxiv.org/abs/2408.12569](https://arxiv.org/abs/2408.12569)
- **Demo:** [Sapiens Gradio Spaces](https://huggingface.co/collections/facebook/sapiens-66d22047daa6402d565cb2fc)
## Uses
Pose 0.3B model can be used to perform 28 class body part segmentation on human images.