rawalkhirodkar
commited on
Commit
•
6bb2907
1
Parent(s):
48f981f
Update model card for Sapiens with architecture details
Browse files
README.md
ADDED
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
language: en
|
3 |
+
license: cc-by-nc-4.0
|
4 |
+
---
|
5 |
+
|
6 |
+
# Pretrain-Sapiens-2B-Bfloat16
|
7 |
+
|
8 |
+
### Model Details
|
9 |
+
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.
|
10 |
+
Sapiens-2B 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.
|
11 |
+
|
12 |
+
- **Developed by:** Meta
|
13 |
+
- **Model type:** Vision Transformer
|
14 |
+
- **License:** Creative Commons Attribution-NonCommercial 4.0
|
15 |
+
- **Task:** pretrain
|
16 |
+
- **Format:** bfloat16
|
17 |
+
- **File:** sapiens_2b_epoch_660_bfloat16.pt2
|
18 |
+
|
19 |
+
### Model Card
|
20 |
+
- **Image Size:** 1024 x 1024
|
21 |
+
- **Num Parameters:** 2.163 B
|
22 |
+
- **FLOPs:** 8.709 TFLOPs
|
23 |
+
- **Patch Size:** 16 x 16
|
24 |
+
- **Embedding Dimensions:** 1920
|
25 |
+
- **Num Layers:** 48
|
26 |
+
- **Num Heads:** 32
|
27 |
+
- **Feedforward Channels:** 7680
|
28 |
+
|
29 |
+
### More Resources
|
30 |
+
- **Repository:** [https://github.com/facebookresearch/sapiens](https://github.com/facebookresearch/sapiens)
|
31 |
+
- **Paper:** [https://arxiv.org/abs/2408.12569](https://arxiv.org/abs/2408.12569)
|
32 |
+
- **Demo:** [Sapiens Gradio Spaces](https://huggingface.co/collections/facebook/sapiens-66d22047daa6402d565cb2fc)
|
33 |
+
- **Project Page:** [https://about.meta.com/realitylabs/codecavatars/sapiens](https://about.meta.com/realitylabs/codecavatars/sapiens/)
|
34 |
+
- **Additional Results:** [https://rawalkhirodkar.github.io/sapiens](https://rawalkhirodkar.github.io/sapiens/)
|
35 |
+
- **HuggingFace Collection:** [https://huggingface.co/collections/facebook/sapiens-66d22047daa6402d565cb2fc](https://huggingface.co/collections/facebook/sapiens-66d22047daa6402d565cb2fc)
|
36 |
+
|
37 |
+
## Uses
|
38 |
+
Pretrained 2B model can be used for feature extraction, fine-tuning, or as a starting point for training new models.
|