Models by task
Every family the library ships, generated from the v1.7.0 model registry: 115 families across 22 tasks.
Detection
47 familiesBoxes around objects. The task most of the library is built for.
s, m, l, x at 640 px
2b, 4b at 1440 px
Segmentation
15 familiesPer-instance masks rather than boxes.
2b, 4b at 1440 px
Keypoints
7 familiesSkeletons and landmarks on each detected instance.
Classification
15 familiesOne label for the whole image, and the backbones behind everything else.
Oriented boxes
4 familiesRotated boxes, for aerial imagery and anything that does not sit axis-aligned.
Depth
8 familiesDistance per pixel from a single image.
Semantic segmentation
10 familiesA class for every pixel, without separating instances.
Points and counting
9 familiesCentroids instead of boxes, cheap enough for microcontrollers.
2b at 1344 px
2b, 4b at 1440 px
Robot policies
1 familiesAction chunks from camera observations and robot state.
smolvla: base at 512 px; act_policy: base at 224 px; diffusion_policy: base at 224 px