rawalkhirodkar commited on
Commit
6bb2907
1 Parent(s): 48f981f

Update model card for Sapiens with architecture details

Browse files
Files changed (1) hide show
  1. README.md +38 -0
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.