yolov3-tiny-onnx-TensorRT icon indicating copy to clipboard operation
yolov3-tiny-onnx-TensorRT copied to clipboard

run onnx_to_tensorrt.py

Open yinjiayang opened this issue 5 years ago • 0 comments

line 211( trt_outputs = [output.reshape(shape) for output, shape in zip(trt_outputs, output_shapes)]), in main trt_outputs = [output.reshape(shape) for output, shape in zip(trt_outputs, output_shapes)] ValueError: cannot reshape array of size 43095 into shape (1,18,13,13)

yinjiayang avatar Dec 07 '20 02:12 yinjiayang