Update onnx_test.py (#2)

- Update onnx_test.py (675b74940c2cb69557dca75d986203e75716d3a3)


Co-authored-by: yixiong huo <yixionghuo@users.noreply.huggingface.co>

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)