Add RT-DETR

utils/export_rtdetr_pytorch.py

@@ -22,17 +22,6 @@ class DeepStreamOutput(nn.Module):
         return boxes, scores, classes
 
 
-class DeepStreamInput(nn.Module):
-    def __init__(self, img_size, device):
-        self.img_size = img_size
-        self.device = device
-        super().__init__()
-
-    def forward(self, x):
-        size = torch.tensor([[*self.img_size]]).to(self.device)
-        return x, size
-
-
 def suppress_warnings():
     warnings.filterwarnings('ignore', category=torch.jit.TracerWarning)
     warnings.filterwarnings('ignore', category=UserWarning)