Error: número de elemento de entrada (408000) número de elemento reformado (102000)

Se informa un error al ejecutar la demostración de Tengine en rv1126,

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.

El comando utilizado es

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

Después de buscar, la razón es que hay el siguiente código en el código tm_yolov5s.cpp

No hay una capa de enfoque en la versión 6 de yolov5. La capa de enfoque se reemplaza por convolución (la capa de enfoque no es compatible con algunos hardware), y el modelo aquí no es la versión 6.

Cambie #define YOLOV5_VERSION6 1 a #define YOLOV5_VERSION6 0.