Merge pull request #1448 from roboflow/add_rfdetr_aliases

rf-detr aliases

Author: Matvezy

inference/core/version.py

@@ -1,4 +1,4 @@
-__version__ = "0.51.6"
+__version__ = "0.51.7"
 
 
 if __name__ == "__main__":

inference/models/aliases.py

@@ -55,6 +55,9 @@
 RFDETR_ALIASES = {
     "rfdetr-base": "coco/36",
     "rfdetr-large": "coco/37",
+    "rfdetr-nano": "coco/38",
+    "rfdetr-small": "coco/39",
+    "rfdetr-medium": "coco/40",
 }
 
 REGISTERED_ALIASES = {

inference/models/rfdetr/rfdetr.py

@@ -6,7 +6,6 @@
 import cv2
 import numpy as np
 import onnxruntime
-import torchvision.transforms.functional
 from PIL import Image
 
 from inference.core.entities.requests.inference import InferenceRequestImage
@@ -38,8 +37,6 @@
 
 if USE_PYTORCH_FOR_PREPROCESSING:
     import torch
-    import torchvision
-    from torchvision.transforms.functional import pil_to_tensor
 
     CUDA_IS_AVAILABLE = torch.cuda.is_available()
 
@@ -374,6 +371,9 @@ def postprocess(
                     topk_labels,
                 )
             )
+            batch_predictions = batch_predictions[
+                batch_predictions[:, 6] < len(self.class_names)
+            ]
 
             processed_predictions.append(batch_predictions)
 

inference_sdk/http/utils/aliases.py

@@ -24,6 +24,9 @@
 RFDETR_ALIASES = {
     "rfdetr-base": "coco/36",
     "rfdetr-large": "coco/37",
+    "rfdetr-nano": "coco/38",
+    "rfdetr-small": "coco/39",
+    "rfdetr-medium": "coco/40",
 }
 
 QWEN_ALIASES = {