Skip to content

Commit e436ae8

Browse files
Add xl rfdetrs types (#1937)
* model types * add new types * add new types --------- Co-authored-by: Grzegorz Klimaszewski <166530809+grzegorz-roboflow@users.noreply.github.com>
1 parent 39d9759 commit e436ae8

File tree

2 files changed

+13
-1
lines changed
  • inference_models/inference_models/models/auto_loaders
  • inference/models

2 files changed

+13
-1
lines changed

inference/models/utils.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,17 +122,20 @@
122122
("object-detection", "yolo26l"): YOLO26ObjectDetection,
123123
("object-detection", "yolo26x"): YOLO26ObjectDetection,
124124
("object-detection", "rfdetr-base"): RFDETRObjectDetection,
125-
("object-detection", "rfdetr-large"): RFDETRObjectDetection,
126125
("object-detection", "rfdetr-nano"): RFDETRObjectDetection,
127126
("object-detection", "rfdetr-small"): RFDETRObjectDetection,
128127
("object-detection", "rfdetr-medium"): RFDETRObjectDetection,
128+
("object-detection", "rfdetr-large"): RFDETRObjectDetection,
129+
("object-detection", "rfdetr-xlarge"): RFDETRObjectDetection,
130+
("object-detection", "rfdetr-2xlarge"): RFDETRObjectDetection,
129131
("instance-segmentation", "rfdetr-seg-preview"): RFDETRInstanceSegmentation,
130132
("instance-segmentation", "rfdetr-seg-nano"): RFDETRInstanceSegmentation,
131133
("instance-segmentation", "rfdetr-seg-small"): RFDETRInstanceSegmentation,
132134
("instance-segmentation", "rfdetr-seg-medium"): RFDETRInstanceSegmentation,
133135
("instance-segmentation", "rfdetr-seg-large"): RFDETRInstanceSegmentation,
134136
("instance-segmentation", "rfdetr-seg-xlarge"): RFDETRInstanceSegmentation,
135137
("instance-segmentation", "rfdetr-seg-xxlarge"): RFDETRInstanceSegmentation,
138+
("instance-segmentation", "rfdetr-seg-2xlarge"): RFDETRInstanceSegmentation,
136139
(
137140
"instance-segmentation",
138141
"yolov11n",

inference_models/inference_models/models/auto_loaders/core.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,16 @@
8787
"rfdetr-medium",
8888
"rfdetr-nano",
8989
"rfdetr-large",
90+
"rfdetr-xlarge",
91+
"rfdetr-2xlarge",
9092
"rfdetr-seg-preview",
93+
"rfdetr-seg-nano",
94+
"rfdetr-seg-small",
95+
"rfdetr-seg-medium",
96+
"rfdetr-seg-large",
97+
"rfdetr-seg-xlarge",
98+
"rfdetr-seg-2xlarge",
99+
"rfdetr-seg-xxlarge",
91100
}
92101

93102
DEFAULT_KWARGS_PARAMS_TO_BE_FORWARDED_TO_DEPENDENT_MODELS = [

0 commit comments

Comments
 (0)