Add support for RF-DETR models (#17298)

* Add support for rf-detr models

* Add docs for rf-detr model

* Cleanup
This commit is contained in:
Nicolas Mowen
2025-03-21 19:55:46 -05:00
committed by GitHub
parent 4e83237d47
commit 48e4c44b32
4 changed files with 97 additions and 5 deletions
+2 -1
View File
@@ -33,11 +33,12 @@ class InputDTypeEnum(str, Enum):
class ModelTypeEnum(str, Enum):
dfine = "dfine"
rfdetr = "rfdetr"
ssd = "ssd"
yolox = "yolox"
yolov9 = "yolov9"
yolonas = "yolonas"
dfine = "dfine"
yologeneric = "yolo-generic"