Detection models
14 detection models. Each section says what the model is, whether it may be used commercially and how fast it is; licences and checks are under Details.
Input: one RGB image. Output: boxes, class labels, scores and instance masks.
Measured on RTX 5060 Ti, torch 2.14.1+cu130; fastforks bench MODEL --size HxW measures yours. The summary uses one input, 800x1344: kernels time with its speed-up over the faster original run, and kernels peak reserved memory with its ratio to the smaller original one. Where measured (one variant per family), a variant's table also lists portable under torch.compile and the fp16 TensorRT engine of portable (export; memory: the engine's device memory). – = not measured.
Other devices: Apple M1 Pro 16 GB, NVIDIA L4, NVIDIA L40S, NVIDIA A10, NVIDIA A100-SXM4-40GB, NVIDIA RTX PRO 6000; across devices.
| model | commercial | kernels time (ms), speed-up vs original | kernels peak memory (MiB, reserved), ratio to original |
|---|---|---|---|
detr-r101-coco | yes | 13.69, 5.8x | 328, 0.52x |
detr-r50-coco | yes | 9.93, 5.6x | 272, 0.5x |
dfine-l-coco | yes | 8.695, 4.3x | 300, 0.91x |
dfine-m-coco | yes | 5.966, 5.2x | 252, 0.98x |
dfine-n-coco | yes | 2.723, 8.9x | 202, 1.7x |
dfine-s-coco | yes | 4.037, 6.4x | 212, 1.2x |
dfine-x-coco | yes | 14.086, 3.2x | 364, 0.75x |
faster-rcnn-r50-fpn-3x-coco | yes | 19.47, 2.1x | 450, 0.64x |
mask-rcnn-r50-fpn-3x-coco | yes | 20.98, 2.1x | 400, 0.62x |
rtdetrv2-l-coco | yes | 6.61, 4.6x | 306, 0.74x |
rtdetrv2-m-coco | yes | 5.27, 3.2x | 274, 1.1x |
rtdetrv2-ms-coco | yes | 5.46, 4.1x | 306, 0.85x |
rtdetrv2-s-coco | yes | 4.19, 3.4x | 250, 1.1x |
rtdetrv2-x-coco | yes | 9.24, 3.9x | 384, 0.64x |
All models
Modes, licences, training data and outputs per model
Commercial status per variant: yes if the code and the weights licence both allow commercial use (the training data is listed but does not change it; weights without a licence of their own are assumed to carry the code licence); no if either forbids it; unclear for custom licence terms (treat as no).
| model | modes | code licence | weights licence | training data | commercial | outputs |
|---|---|---|---|---|---|---|
detr-r101-coco | portable, kernels | Apache-2.0 | Apache-2.0 | coco, imagenet-1k | yes | boxes, scores, labels |
detr-r50-coco | portable, kernels | Apache-2.0 | Apache-2.0 | coco, imagenet-1k | yes | boxes, scores, labels |
dfine-l-coco | portable, kernels | Apache-2.0 | Apache-2.0 | coco, imagenet-1k | yes | boxes, scores, labels |
dfine-m-coco | portable, kernels | Apache-2.0 | Apache-2.0 | coco, imagenet-1k | yes | boxes, scores, labels |
dfine-n-coco | portable, kernels | Apache-2.0 | Apache-2.0 | coco, imagenet-1k | yes | boxes, scores, labels |
dfine-s-coco | portable, kernels | Apache-2.0 | Apache-2.0 | coco, imagenet-1k | yes | boxes, scores, labels |
dfine-x-coco | portable, kernels | Apache-2.0 | Apache-2.0 | coco, imagenet-1k | yes | boxes, scores, labels |
faster-rcnn-r50-fpn-3x-coco | portable, kernels | Apache-2.0 | CC-BY-SA-3.0 | coco | yes | boxes, scores, labels, masks |
mask-rcnn-r50-fpn-3x-coco | portable, kernels | Apache-2.0 | CC-BY-SA-3.0 | coco | yes | boxes, scores, labels, masks |
rtdetrv2-l-coco | portable, kernels | Apache-2.0 | Apache-2.0 (assumed from code) | coco, imagenet-1k | yes | boxes, scores, labels |
rtdetrv2-m-coco | portable, kernels | Apache-2.0 | Apache-2.0 (assumed from code) | coco, imagenet-1k | yes | boxes, scores, labels |
rtdetrv2-ms-coco | portable, kernels | Apache-2.0 | Apache-2.0 (assumed from code) | coco, imagenet-1k | yes | boxes, scores, labels |
rtdetrv2-s-coco | portable, kernels | Apache-2.0 | Apache-2.0 (assumed from code) | coco, imagenet-1k | yes | boxes, scores, labels |
rtdetrv2-x-coco | portable, kernels | Apache-2.0 | Apache-2.0 (assumed from code) | coco, imagenet-1k | yes | boxes, scores, labels |
Training datasets
| id | dataset | licence | commercial use |
|---|---|---|---|
coco | COCO 2017 (train, instance annotations) | CC-BY-4.0 | unclear |
imagenet-1k | ImageNet-1k (ILSVRC 2012 classification) | LicenseRef-noncommercial | no |