Update widerface_onnx_evalute.py

widerface_onnx_evalute.py

@@ -67,6 +67,8 @@ def Retinaface_evalute(run_ort, args):
         img_raw = cv2.imread(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