From d3241dbc29e705e9b14af7cfe86fd82c5c02eb23 Mon Sep 17 00:00:00 2001 From: ives-kwy Date: Wed, 29 Sep 2021 16:09:44 +0800 Subject: [PATCH] Fix bug of onnx/yolov5s demo: ValueError: cannot reshape array of size 7225 into shape (40,85,1,1) --- examples/onnx/yolov5/test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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)