Skip to content

Commit 02461c9

Browse files
Merge pull request #1897 from roboflow/rfdetr-large-and-seg-sizes
Add new RF-DETR-Seg sizes
2 parents bdc0d78 + a710173 commit 02461c9

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

inference/models/utils.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,12 @@
117117
("object-detection", "rfdetr-small"): RFDETRObjectDetection,
118118
("object-detection", "rfdetr-medium"): RFDETRObjectDetection,
119119
("instance-segmentation", "rfdetr-seg-preview"): RFDETRInstanceSegmentation,
120+
("instance-segmentation", "rfdetr-seg-nano"): RFDETRInstanceSegmentation,
121+
("instance-segmentation", "rfdetr-seg-small"): RFDETRInstanceSegmentation,
122+
("instance-segmentation", "rfdetr-seg-medium"): RFDETRInstanceSegmentation,
123+
("instance-segmentation", "rfdetr-seg-large"): RFDETRInstanceSegmentation,
124+
("instance-segmentation", "rfdetr-seg-xlarge"): RFDETRInstanceSegmentation,
125+
("instance-segmentation", "rfdetr-seg-xxlarge"): RFDETRInstanceSegmentation,
120126
(
121127
"instance-segmentation",
122128
"yolov11n",

0 commit comments

Comments
 (0)