Sapiens segmentation

• Original code: https://github.com/facebookresearch/sapiens at 2cb07227a740
• Code licence: CC-BY-NC-4.0 (noncommercial); files: LICENSE
• Weights licence: CC-BY-NC-4.0
• Licence evidence: https://huggingface.co/facebook/sapiens-seg-1b-torchscript/blob/584504252a6d2a27ea6560927811db734ee42332/README.md
• Training data: MAE pretraining on Humans-300M (Meta-internal, not released); segmentation fine-tuning on Goliath body-part annotations (Meta, not released)
• Kernel libraries: fastbb, cudnn_hacc, seg_up
• Variants: sapiens-seg-0.3b-goliath, sapiens-seg-1b-goliath
• Outputs: logits; label_set: goliath-28
• Input size: any H x W >= 32, padded to a multiple of 32
• Tested up to (portable, fp16, RTX 5060 Ti 16 GB): sapiens-seg-0.3b-goliath 3000x4000 (2.2 GiB peak allocated, 3.2 reserved); sapiens-seg-1b-goliath 3000x4000 (3.8 GiB peak allocated, 5.0 reserved)
• sapiens-seg-0.3b-goliath 1024x2048, first call: portable + torch.compile 12 s (cold Inductor cache), TensorRT fp16 53 s (ONNX + engine).
• sapiens-seg-0.3b-goliath original 1024x2048: torch.autocast(fp16) around the forward
• sapiens-seg-1b-goliath original 1024x2048: torch.autocast(fp16) around the forward
• sapiens-seg-0.3b-goliath load / first call: original 4.92 s / 1.024 s (512x512), portable 1.22 s / 0.132† s, kernels 0.96† s / 0.092† s
• sapiens-seg-1b-goliath load / first call: original 11.33 s / 0.569 s (512x512), portable 3.54 s / 0.274† s, kernels 2.27† s / 0.18† s
• † measured in the speed sweep after the portable load (or an earlier size) in the same process: file cache and kernels are warm, so the time is lower than a fresh start.
• Protocol, first call, 3 warm-up, peak reset, median of 3 rounds x median of 5 (tools/speed_check.py), synthetic CUDA input (bench_inputs); peak = max over the first call and the timed calls

Measured

Any body part counts as person; ground truth = the COCO person masks. Original = the original code named under Details, same checkpoint and protocol.

Measured on NVIDIA GeForce RTX 5060 Ti, torch 2.14.1+cu130, fastforks 939adfe8, 2026-11-06.

Parity

CPU, float32: max |diff| <= 0.002 vs the original. GPU, float16 (the default), portable and kernels: mean |diff| <= 0.05 and p99 <= 0.25 (absolute error of the output logits).