update infer code for NCHW->NHWC

infer_onnx.py

@@ -139,8 +139,10 @@ if __name__ == '__main__':
         providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
         provider_options = None
     session = ort.InferenceSession(args.model, providers=providers, provider_options=provider_options)
-    ort_inputs = {session.get_inputs()[0].name: img[None, :, :, :]}
+    # ort_inputs = {session.get_inputs()[0].name: img[None, :, :, :]}
+    ort_inputs = {session.get_inputs()[0].name: np.transpose(img[None, :, :, :], (0, 2 ,3, 1))}
     outputs = session.run(None, ort_inputs)
+    outputs = [np.transpose(out, (0, 3, 1, 2)) for out in outputs]
     dets = postprocess(outputs, input_shape, ratio)
     if dets is not None:
         final_boxes, final_scores, final_cls_inds = dets[:, :4], dets[:, 4], dets[:, 5]