在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即可。