Tensorrt-Yolov3-tiny icon indicating copy to clipboard operation
Tensorrt-Yolov3-tiny copied to clipboard

Cannot convert from yolov3-tiny to onnx

Open bhavitvyamalik opened this issue 4 years ago • 0 comments

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.

bhavitvyamalik avatar May 12 '20 16:05 bhavitvyamalik