Error: input elem num(408000) reshaped elem num(102000)

在rv1126上运行Tengine的demo时报错,

Error: input elem num(408000) != reshaped elem num(102000)
Tengine FATAL: Infer node(id: 276, op: Reshape) shape failed.
Tengine: Infer shape of graph failed(-1).
Prerun multithread graph failed.

使用的命令为

./build/install/bin/tm_yolov5s -m models/yolov5s.tmfile -i images/ssd_dog.jpg -r 1 -t 1

经查找,原因是代码tm_yolov5s.cpp中有如下代码

yolov5的version6中没有focus层了,focus层用卷积代替了(focus层有些硬件不支持),我们这里的模型并不是version6版本。

将#define YOLOV5_VERSION6 1修改为#define YOLOV5_VERSION6 0即可。

猜你喜欢

转载自blog.csdn.net/u013171226/article/details/122982661
num