Update README.md for NCHW->NHWC

README.md

@@ -76,8 +76,12 @@ Then unzip the files and move them to the following directories (or create soft
       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, :, :, :]}
+  # NCHW format
+  # ort_inputs = {session.get_inputs()[0].name: img[None, :, :, :]}
+  # NHWC format
+  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] # for NHWC format
   dets = postprocess(outputs, input_shape, ratio)
   if dets is not None:
       final_boxes, final_scores, final_cls_inds = dets[:, :4], dets[:, 4], dets[:, 5]