Update widerface_onnx_evalute.py (#2)

- Update widerface_onnx_evalute.py (d66a469eb68d0be579448d73a2b0f6a8389a365b)


Co-authored-by: Hongyang Wei <[email protected]>

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