Add RT-DETR

utils/export_rtdetr_pytorch.py

@@ -20,17 +20,6 @@ class DeepStreamOutput(nn.Module):
         scores, classes = torch.max(x['pred_logits'], 2, keepdim=True)
         classes = classes.float()
         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():