Merge pull request #35 from rsnk96/patch-1

Fix exporting onnx file that can only be used for inference with max batch size of 1

Author: Peterande

tools/deployment/export_onnx.py

@@ -51,7 +51,7 @@ def forward(self, images, orig_target_sizes):
 
     model = Model()
 
-    data = torch.rand(1, 3, 640, 640)
+    data = torch.rand(32, 3, 640, 640)
     size = torch.tensor([[640, 640]])
     _ = model(data, size)