Update onnx_test.py

onnx_test.py

@@ -962,8 +962,12 @@ def test(data,
         whwh = torch.Tensor([width, height, width, height]).to(device)
 
         if onnx_runtime:
+            # outputs = onnx_model.run(
+            #     None, {onnx_model.get_inputs()[0].name: imgs.cpu().numpy()})
             outputs = onnx_model.run(
-                None, {onnx_model.get_inputs()[0].name: imgs.cpu().numpy()})
+                None, {onnx_model.get_inputs()[0].name: np.transpose(imgs.cpu().numpy(), (0, 2, 3, 1))})
+            outputs = [np.transpose(out, (0, 3, 1, 2)) for out in outputs]
+      
             outputs = [torch.tensor(item).to(device) for item in outputs]
             inf_out, train_out = post_process(outputs)