sapiens2-seg-5b

Sapiens2-5B-Body-Part Per-pixel body-part segmentation with 29 classes (28 parts + background). This repository contains the **5B Body-Part Segmentation** checkpoint, finetuned from the Sapiens2-5B pretrained backbone. 📄 **Paper:** arXiv:2604.21681 🌐 **Project Page:** rawalkhirodkar.github.io/sapiens2 💻 **Code:** github.com/facebookresearch/sapiens2 Model Details **Developed by:** Meta **Model type:** Vision Transformer **License:** Sapiens2 License **Task:** seg **Base model:** facebook/sapiens2-pretrain-5b **Format:** safetensors **File:** `sapiens2_5b_seg.safetensors` Quick Start Install the Sapiens2 repo (`pip install -e .`), download the checkpoint, and run the demo: See the Body-Part Segmentation guide for details on inputs, outputs, and visualization options. Model Card | Field | Value | |-------|-------| | Architecture | Sapiens2 ViT backbone + Body-Part Segmentation head | | Backbone parameters | 5.071 B | | Backbone FLOPs | 15.722 T | | Embedding dim | 2432 | | Layers | 56 | | Attention heads | 32 | | Inference resolution | 1024 × 768 (H × W) | | Patch size | 16 | Sapiens2-Body-Part Family | Model | Params | FLOPs | Embed dim | Layers | Heads | |-------|--------|-------|-----------|--------|-------| | Sapiens2-0.4B | 0.398 B | 1.260 T | 1024 | 24 | 16 | | Sapiens2-0.8B | 0.818 B | 2.592 T | 1280 | 32 | 16 | | Sapiens2-1B | 1.462 B | 4.715 T | 1536 | 40 | 24 | | **Sapiens2-5B** *(this)* | 5.071 B | 15.722 T | 2432 | 56 | 32 | See the Sapiens2 Collection for all variants and other downstream task checkpoints. Intended Use Body-Part Segmentation on human-centric imagery Research on human-centric vision License Released under the Sapiens2 License. Citation