update eval code for NCHW->NHWC

eval_onnx.py

@@ -93,7 +93,9 @@ class COCOEvaluator:
             is_time_record = cur_iter < len(self.dataloader) - 1
             if is_time_record:
                 start = time.time()
-            outputs = ort_sess.run(None, {input_name: imgs.numpy()})
+            # outputs = ort_sess.run(None, {input_name: imgs.numpy()})
+            outputs = ort_sess.run(None, {input_name: np.transpose(imgs.numpy(), (0, 2, 3, 1))})
+            outputs = [np.transpose(out, (0, 3, 1, 2)) for out in outputs]
             outputs = [torch.Tensor(out) for out in outputs]
             outputs = head_postprocess(outputs)
             if is_time_record: