Hello,
When converting a .pb graph to .uff using convert-to-uff
, I get the following error which appears to be linked to a depth-wise convolution layer:
Converting to UFF graph
Traceback (most recent call last):
File "/opt/tools/anaconda2/bin/convert-to-uff", line 11, in <module>
sys.exit(main())
File "/opt/tools/anaconda2/lib/python2.7/site-packages/uff/bin/convert_to_uff.py", line 79, in main
output_filename=args.output
File "/opt/tools/anaconda2/lib/python2.7/site-packages/uff/converters/tensorflow/conversion_helpers.py", line 159, in from_tensorflow_frozen_model
return from_tensorflow(graphdef, output_nodes, preprocessor, **kwargs)
File "/opt/tools/anaconda2/lib/python2.7/site-packages/uff/converters/tensorflow/conversion_helpers.py", line 132, in from_tensorflow
name="main")
File "/opt/tools/anaconda2/lib/python2.7/site-packages/uff/converters/tensorflow/converter.py", line 77, in convert_tf2uff_graph
uff_graph, input_replacements)
File "/opt/tools/anaconda2/lib/python2.7/site-packages/uff/converters/tensorflow/converter.py", line 64, in convert_tf2uff_node
op, name, tf_node, inputs, uff_graph, tf_nodes=tf_nodes)
File "/opt/tools/anaconda2/lib/python2.7/site-packages/uff/converters/tensorflow/converter.py", line 43, in convert_layer
return cls.registry_[op](name, tf_node, inputs, uff_graph, **kwargs)
File "/opt/tools/anaconda2/lib/python2.7/site-packages/uff/converters/tensorflow/converter_functions.py", line 276, in convert_depthwise_conv2d_native
return _conv2d_helper(name, tf_node, inputs, uff_graph, func="depthwise", **kwargs)
File "/opt/tools/anaconda2/lib/python2.7/site-packages/uff/converters/tensorflow/converter_functions.py", line 301, in _conv2d_helper
number_groups = int(wt.attr['value'].tensor.tensor_shape.dim[2].size)
IndexError: list index out of range
Note that I am using python 2.7, tensorflow 1.11 (to create the .pb file) and TensorRT 5.0.0.10 (which comes with uff 0.5.1) on a p100 GPU.
I can provide the .pb file if needed, it is quite small (900kb)
Any help would be very much appreciated.
Thanks