Update widerface_onnx_inference.py

widerface_onnx_inference.py

@@ -61,6 +61,8 @@ def Retinaface_inference(run_ort, args):
     img_raw = cv2.imread(args.image_path, cv2.IMREAD_COLOR)
     # preprocess
     img, scale, resize = preprocess(img_raw, INPUT_SIZE, DEVICE)
+    # to NHWC
+    img = np.transpose(img, (0, 2, 3, 1))
     # forward 
     outputs = run_ort.run(None, {run_ort.get_inputs()[0].name: img})
     # postprocess