sapiens2-seg-0.8b

Sapiens2-0.8B-Body-Part

Per-pixel body-part segmentation with 29 classes (28 parts + background).

This repository contains the 0.8B Body-Part Segmentation checkpoint, finetuned from the Sapiens2-0.8B 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-0.8b
• Format: safetensors
• File: sapiens2_0.8b_seg.safetensors

Quick Start

Install the Sapiens2 repo (pip install -e .), download the checkpoint, and run the demo:

# 1. Download the checkpoint to $SAPIENS_CHECKPOINT_ROOT/seg/
hf download facebook/sapiens2-seg-0.8b sapiens2_0.8b_seg.safetensors \
    --local-dir ~/sapiens2_host/seg

# 2. Run the demo (edit INPUT, OUTPUT, and MODEL_NAME inside the script)
cd $SAPIENS_ROOT/sapiens/dense
./scripts/demo/seg.sh

See the Body-Part Segmentation guide for details on inputs, outputs, and visualization options.

Model Card

Sapiens2-Body-Part Family

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

@article{khirodkarsapiens2,
  title={Sapiens2},
  author={Khirodkar, Rawal and Wen, He and Martinez, Julieta and Dong, Yuan and Su, Zhaoen and Saito, Shunsuke},
  journal={arXiv preprint arXiv:2604.21681},
  year={2026}
}