run onnx_to_tensorrt.py

zombie0117 / yolov3-tiny-onnx-TensorRT

convert your yolov3-tiny model to trt model

90 stars 34 forks source link

run onnx_to_tensorrt.py #19

Open yinjiayang opened 3 years ago

yinjiayang commented 3 years ago

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)