Downloads · 30 days
10.6K
13% of all-time downloads
sapiens2-seg-0.4b
facebook/sapiens2-seg-0.4b
sapiens2-seg-0.4b is a image segmentation model from facebook. Use it for the image segmentation task on the model card, and read the license before you ship it in a product. It is set up for sapiens2. The card lists the license as other.
Per-pixel body-part segmentation with 29 classes (28 parts + background).
All-time downloads
79.1K
Public
Parameters
407M
1.6 GB on disk
Likes
7
Public
Repo makeup
Click a slice to open those files.
.safetensors3.3 GB · 100%
Parameter types
How the weights are stored.
F32407M · 100%
Base models
Model card
From the Hugging Face model README
Sapiens2-0.4B-Body-Part
Per-pixel body-part segmentation with 29 classes (28 parts + background).
This repository contains the 0.4B Body-Part Segmentation checkpoint, finetuned from the Sapiens2-0.4B 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.4b
- Format: safetensors
- File:
sapiens2_0.4b_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.4b sapiens2_0.4b_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
| Field | Value |
|---|---|
| Architecture | Sapiens2 ViT backbone + Body-Part Segmentation head |
| Backbone parameters | 0.398 B |
| Backbone FLOPs | 1.260 T |
| Embedding dim | 1024 |
| Layers | 24 |
| Attention heads | 16 |
| Inference resolution | 1024 × 768 (H × W) |
| Patch size | 16 |
Sapiens2-Body-Part Family
| Model | Params | FLOPs | Embed dim | Layers | Heads |
|---|---|---|---|---|---|
| Sapiens2-0.4B (this) | 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 | 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
@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}
}
- Task
- Image Segmentation
- Library
- sapiens2
- Type
- sapiens2
- License
- other
- Created
- Apr 23, 2026
- Updated
- Jun 15, 2026