Tensorrt-Yolov3-tiny
Tensorrt-Yolov3-tiny copied to clipboard
Cannot convert from yolov3-tiny to onnx
After going through this issue I got another issue while converting yolov3-tiny to onnx.
Traceback (most recent call last):
File "yolov3_to_onnx.py", line 832, in <module>
main()
File "yolov3_to_onnx.py", line 820, in main
verbose=True)
File "yolov3_to_onnx.py", line 420, in build_onnx_graph
major_node_specs = self._make_onnx_node(layer_name, layer_dict)
File "yolov3_to_onnx.py", line 474, in _make_onnx_node
layer_name, layer_dict)
File "yolov3_to_onnx.py", line 514, in _make_input_tensor
batch_size, channels, height, width])
File "/home/jetson-beta/onnx-tensorrt/local/lib/python2.7/site-packages/onnx/helper.py", line 351, in make_tensor_value_info
'Needs to of integer_types or text_type.'.format(d))
ValueError: Invalid item in shape: 1. Needs to of integer_types or text_type.