trt_outputs = [output.reshape(shape) for output, shape ValueError: cannot reshape array of size 20577 into shape (1,255,19,19) about tensorrt_demos

[jkjung-avt]

Please refer to this post: #51 (comment)

20577 / 19 / 19 = 57
(14 + 5) * 3 = 57

I think your custom YOLOv3 model is detecting 14 classes of objects. So you should modify "category_num" from 80 to 14, and "output_shapes" from 255 to 57.

[chenyuqing]

It works, Thaks a lot ! ~ @jkjung-avt

[jkjung-avt]

You are welcome.

[jkjung-avt]

I've added a "--category_num" command-line option to make it easier to adapt my TensorRT YOLOv3 code to custom trained models. Please check out my blog post TensorRT YOLOv3 For Custom Trained Models for details.