From 5af9da189d91fa8808695c488a417f9f1c920b77 Mon Sep 17 00:00:00 2001
From: Marcos Luciano <marcoslucianops@gmail.com>
Date: Wed, 1 Nov 2023 18:42:56 -0300
Subject: [PATCH] Add RT-DETR

---
 utils/export_rtdetr_pytorch.py | 11 -----------
 1 file changed, 11 deletions(-)

diff --git a/utils/export_rtdetr_pytorch.py b/utils/export_rtdetr_pytorch.py
index 140cc71..600dd44 100755
--- a/utils/export_rtdetr_pytorch.py
+++ b/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():