Toybrick

标题: yolov5 test.py "cannot reshape array"问题求助 [打印本页]

作者: nbplus 时间: 2022-9-15 15:11
标题: yolov5 test.py "cannot reshape array"问题求助
rknn-toolkit-1.7.1，example中的pytorch yolov5 test.py，实际测试出现如下问题，请大牛们指点，搞了两天了都没搞定，谢谢
--> Init runtime environment
librknn_runtime version 1.7.1 (bd41dbc build: 2021-10-28 16:15:23 base: 1131)
done
--> Running model
Warning: Cmodel got negative postshift!
Warning: Cmodel got negative postshift!
Warning: Cmodel got negative postshift!
Traceback (most recent call last):
File "test.py", line 294, in <module>
input1_data = input1_data.reshape([3,-1]+list(input1_data.shape[-2:]))
ValueError: cannot reshape array of size 409600 into shape (3,newaxis,40,40)

欢迎光临 Toybrick (https://t.rock-chips.com/)