yolov5-onnx-tensorrt
yolov5-onnx-tensorrt copied to clipboard
The Post processing seems like wrong?
It seems like the post processing shows an error. any ideas?
trtbin coco_yolov5-16.trt
[TensorRT] WARNING: Using an engine plan file across different models of devices is not recommended and is likely to affect performance or even cause errors.
image arg bus.jpg
(1, 3, 608, 608)
execution time: 0.07985496520996094
(3326976,)
Traceback (most recent call last):
File "demo_batch.py", line 87, in <module>
main()
File "demo_batch.py", line 79, in main
output, image_raw_batch = processor.detect(img_batch)
File "/home/administrator/yolov5-onnx-tensorrt/demo/Processor_Batch.py", line 92, in detect
out = out.reshape(self.output_shapes[i])
ValueError: cannot reshape array of size 831744 into shape (1,3,80,80,85)