Skip to content

Commit 0dd8946

Browse files
authored
Merge pull request #1446 from roboflow/new_rfdetr_sizes
Add new model size to rf-detr
2 parents 9b1606b + 3439849 commit 0dd8946

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

inference/models/utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,9 @@
105105
("object-detection", "yolov12x"): YOLOv12ObjectDetection,
106106
("object-detection", "rfdetr-base"): RFDETRObjectDetection,
107107
("object-detection", "rfdetr-large"): RFDETRObjectDetection,
108+
("object-detection", "rfdetr-nano"): RFDETRObjectDetection,
109+
("object-detection", "rfdetr-small"): RFDETRObjectDetection,
110+
("object-detection", "rfdetr-medium"): RFDETRObjectDetection,
108111
(
109112
"instance-segmentation",
110113
"yolov11n",

0 commit comments

Comments
 (0)