CoolFace
Modelpublic

facebook/sapiens2-pose-0.4b

sourceHugging Faceotherupdated 3mo agoView on Hugging Face
10likes2.3kdownloads
Model Card

Sapiens2-0.4B-Pose

308-keypoint top-down pose estimation including detailed face (274 keypoints), hand, and foot keypoints. Predictions follow the Sociopticon keypoint format.

This repository contains the 0.4B Pose Estimation checkpoint, finetuned from the Sapiens2-0.4B pretrained backbone.

Pose is top-down — it requires bounding boxes from a person detector. We use RTMDet.

Model Details

Quick Start

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

bash
# 1. Download the checkpoint to $SAPIENS_CHECKPOINT_ROOT/pose/
hf download facebook/sapiens2-pose-0.4b sapiens2_0.4b_pose.safetensors \
    --local-dir ~/sapiens2_host/pose

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

See the Pose Estimation guide for details on inputs, outputs, and visualization options.

Model Card

FieldValue
ArchitectureSapiens2 ViT backbone + Pose Estimation head
Backbone parameters0.398 B
Backbone FLOPs1.260 T
Embedding dim1024
Layers24
Attention heads16
Inference resolution1024 × 768 (H × W)
Patch size16

Sapiens2-Pose Family

ModelParamsFLOPsEmbed dimLayersHeads
Sapiens2-0.4B (this)0.398 B1.260 T10242416
Sapiens2-0.8B0.818 B2.592 T12803216
Sapiens2-1B1.462 B4.715 T15364024
Sapiens2-5B5.071 B15.722 T24325632

See the Sapiens2 Collection for all variants and other downstream task checkpoints.

Intended Use

  • Pose Estimation on human-centric imagery
  • Research on human-centric vision

License

Released under the Sapiens2 License.

Citation

bibtex
@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}
}