diff --git a/examples/onnx/yolov5/test.py b/examples/onnx/yolov5/test.py
index 817225b..8b450d0 100644
--- a/examples/onnx/yolov5/test.py
+++ b/examples/onnx/yolov5/test.py
@@ -291,9 +291,9 @@ def letterbox(im, new_shape=(640, 640), color=(0, 0, 0)):
     input2_data = input2_data.reshape(*input2_data.shape[1:])
 
     input_data = list()
-    input_data.append(np.transpose(input0_data, (2, 3, 0, 1)))
-    input_data.append(np.transpose(input1_data, (2, 3, 0, 1)))
-    input_data.append(np.transpose(input2_data, (2, 3, 0, 1)))
+    input_data.append(np.transpose(input0_data, (1, 2, 0, 3)))
+    input_data.append(np.transpose(input1_data, (1, 2, 0, 3)))
+    input_data.append(np.transpose(input2_data, (1, 2, 0, 3)))
 
     boxes, classes, scores = yolov5_post_process(input_data)