CoolFace
Modelpublic

facebook/sapiens2-normal-0.8b

sourceHugging Faceotherupdated 3mo agoView on Hugging Face
3likes516downloads
Model Card

Sapiens2-0.8B-Surface

Per-pixel surface-normal estimation (3-channel unit vectors in camera frame).

This repository contains the 0.8B Surface Normal Estimation checkpoint, finetuned from the Sapiens2-0.8B pretrained backbone.

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/normal/
hf download facebook/sapiens2-normal-0.8b sapiens2_0.8b_normal.safetensors \
    --local-dir ~/sapiens2_host/normal

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

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

Model Card

FieldValue
ArchitectureSapiens2 ViT backbone + Surface Normal Estimation head
Backbone parameters0.818 B
Backbone FLOPs2.592 T
Embedding dim1280
Layers32
Attention heads16
Inference resolution1024 ร— 768 (H ร— W)
Patch size16

Sapiens2-Surface Family

ModelParamsFLOPsEmbed dimLayersHeads
Sapiens2-0.4B0.398 B1.260 T10242416
Sapiens2-0.8B (this)0.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

  • โ€”Surface Normal 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}
}