Provide details on the platforms you are using:
OS : Linux 16.04
GPU : GTX 1080ti
nvidia driver version : 384.130
CUDA version : 9.0
CUDNN version : 7.3.1
Python version : 3.5.2
Tensorflow version : 1.11.0
TensorRT version : 5.0.0.10
Describe the problem
I used TensorRT5 with tensorflow. I have downloaded pre-trained inception v3 model from https://github.com/tensorflow/models/tree/master/research/slim. And made freeze graph model.
when I convert freeze graph to uff, I got error like below
==========================================================================================
attr {
key: “shared_name”
value {
s: “”
}
}
Traceback (most recent call last):
File “/home/hotaek/anaconda3/envs/py352/bin/convert-to-uff”, line 11, in
sys.exit(main())
File “/home/hotaek/anaconda3/envs/py352/lib/python3.5/site-packages/uff/bin/convert_to_uff.py”, line 79, in main
output_filename=args.output
File “/home/hotaek/anaconda3/envs/py352/lib/python3.5/site-packages/uff/converters/tensorflow/conversion_helpers.py”, line 159, in from_tensorflow_frozen_model
return from_tensorflow(graphdef, output_nodes, preprocessor, **kwargs)
File “/home/hotaek/anaconda3/envs/py352/lib/python3.5/site-packages/uff/converters/tensorflow/conversion_helpers.py”, line 132, in from_tensorflow
name=“main”)
File “/home/hotaek/anaconda3/envs/py352/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py”, line 77, in convert_tf2uff_graph
uff_graph, input_replacements)
File “/home/hotaek/anaconda3/envs/py352/lib/python3.5/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 “/home/hotaek/anaconda3/envs/py352/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py”, line 39, in convert_layer
fields = cls.parse_tf_attrs(tf_node.attr)
File “/home/hotaek/anaconda3/envs/py352/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py”, line 209, in parse_tf_attrs
return {key: cls.parse_tf_attr_value(val) for key, val in attrs.items() if val is not None and val.WhichOneof(‘value’) is not None}
File “/home/hotaek/anaconda3/envs/py352/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py”, line 209, in
return {key: cls.parse_tf_attr_value(val) for key, val in attrs.items() if val is not None and val.WhichOneof(‘value’) is not None}
File “/home/hotaek/anaconda3/envs/py352/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py”, line 205, in parse_tf_attr_value
return cls.convert_tf2uff_field(code, val)
File “/home/hotaek/anaconda3/envs/py352/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py”, line 190, in convert_tf2uff_field
‘type’: ‘dtype’, ‘list’: ‘list’}[code]
KeyError: ‘shape’
I thought it cause from unsupported layer. but I can’t find layer has name “shared_name”.
well, I don’t know what’s going on. need help.
Thanks :)
Files