Models by task

Every family the library ships, generated from the v1.5.0 model registry: 82 families across 17 tasks.

Detection

40 families

Boxes around objects. The task most of the library is built for.

⭐RF-DETR

n, s, m, l for detection, pose and oriented boxes; n through xx for segmentation

docs weights

⭐YOLOv9

yolo9: t, s, m, c at 640 px; yolo9_p2: t, s at 640 px

docs weights

D-FINE

n, s, m, l, x at 640 px

docs weights

DEIM

deim: n, s, m, l, x at 640 px

docs weights

EdgeCrafter

s, m, l, x at 640 px

docs weights

RT-DETR

rtdetr: r18, r34, r50, r50m, r101, l, x at 640 px; rtdetrv2: r18, r34, r50, r50m, r101 for detection at 640 px, n, s, m, l, x for oriented boxes at 1024 px; rtdetrv4: s, m, l, x at 640 px

docs weights

YOLO-NAS

s, m, l at 640 px

docs weights

Dome-DETR

s, m, l at 800 px

docs weights

PicoDet

s, m, l at 320 to 640 px

docs weights

RTMDet

t, s, m, l, x at 640 px

docs weights

YOLOv7

b at 640 px

docs weights

YOLOX

n, t, s, m, l, x at 416 to 640 px

docs weights

Florence-2

base, large at 768 px

docs weights

Grounding DINO

t, b at 800 px

docs weights

InternVL3

1b, 2b, 8b at 448 px

docs weights

Kosmos-2

224 at 224 px

docs weights

LFM2-VL

450m at 512 px

docs weights

LibreMODUS

docs weights

LocateAnything

3b at 2500 px

docs weights

OMDet-Turbo

t at 640 px

docs weights

OV-DEIM

s, m, l at 640 px

docs weights

OWLv2

b16, l14 at 960 to 1008 px

docs weights

Qwen3-VL

2b, 4b, 8b at 1024 px

docs weights

SenseNova-Vision

7b at 1024 px

docs weights

SmolVLM2

500m at 512 px

docs weights

CenterNet

resdcn18, dla34 at 512 px

docs weights

Deformable DETR

r50ss, r50ssdc5, r50, r50refine, r50twostage at 800 px

docs weights

DETR

r50, r50dc5, r101, r101dc5 at 800 px

docs weights

DINO-DETR

r50, r50s5, swinl at 800 px

docs weights

EfficientDet

docs weights

Faster R-CNN

n, s, m, l at 320 to 800 px

docs weights

FCOS

r50 at 800 px

docs weights

LW-DETR

t, s, m, l, x at 640 px

docs weights

Mask R-CNN

r50 at 800 px

docs weights

RetinaNet

r50, r50v2 at 800 px

docs weights

SSD

300 at 300 px

docs weights

YOLOv1

t, b at 448 px

docs weights

YOLOv2

t, b at 416 to 608 px

docs weights

YOLOv3

t, b, spp at 416 to 608 px

docs weights

YOLOv4

t, b at 416 to 608 px

docs weights

Segmentation

13 families

Per-instance masks rather than boxes.

⭐RF-DETR

n, s, m, l for detection, pose and oriented boxes; n through xx for segmentation

docs weights

D-FINE

n, s, m, l, x at 640 px

docs weights

EdgeCrafter

s, m, l, x at 640 px

docs weights

RTMDet

t, s, m, l, x at 640 px

docs weights

EdgeTAM

docs weights

MobileSAM

tiny at 1024 px

docs weights

PicoSAM3

pico at 96 px

docs weights

SAM

base, large, huge at 1024 px

docs weights

SAM 2

docs weights

SAM 3

large at 1008 px

docs weights

SenseNova-Vision

7b at 1024 px

docs weights

EoMT

s, b, l at 512 px

docs weights

Mask R-CNN

r50 at 800 px

docs weights

Keypoints

5 families

Skeletons and landmarks on each detected instance.

⭐RF-DETR

n, s, m, l for detection, pose and oriented boxes; n through xx for segmentation

docs weights

EdgeCrafter

s, m, l, x at 640 px

docs weights

YOLO-NAS

s, m, l at 640 px

docs weights

SenseNova-Vision

7b at 1024 px

docs weights

HRNet

docs weights

Classification

12 families

One label for the whole image, and the backbones behind everything else.

ConvNeXt

t, s, b at 224 px

docs weights

DINOv2

n, s, m, l at 518 px

docs weights

EfficientNetV2

b0, b1, b2, b3 at 224 to 300 px

docs weights

MobileNetV4

s, m, l at 224 to 256 px

docs weights

ResNet

18, 34, 50, 101 at 224 px

docs weights

CLIP

docs weights

SigLIP2

docs weights

AlexNet

b at 224 px

docs weights

Swin Transformer

t, s, b, l at 224 px

docs weights

VGG

16, 19, 16bn, 19bn at 224 px

docs weights

ViT

ti, s, b, l at 224 px

docs weights

DeiT

t, s, b at 224 px

docs weights

Oriented boxes

2 families

Rotated boxes, for aerial imagery and anything that does not sit axis-aligned.

⭐RF-DETR

n, s, m, l for detection, pose and oriented boxes; n through xx for segmentation

docs weights

RT-DETR

rtdetr: r18, r34, r50, r50m, r101, l, x at 640 px; rtdetrv2: r18, r34, r50, r50m, r101 for detection at 640 px, n, s, m, l, x for oriented boxes at 1024 px; rtdetrv4: s, m, l, x at 640 px

docs weights