Hi ,
I want to convert ssd model to tensorrt plan file . But I am getting error as
Using output node scores boxes classes
Converting to UFF graph
Traceback (most recent call last):
File "convert_plan.py", line 73, in
data_type
File "convert_plan.py", line 24, in frozenToPlan
text=False,
File "/usr/lib/python2.7/dist-packages/uff/converters/tensorflow/conversion_helpers.py", line 113, in from_tensorflow_frozen_model
return from_tensorflow(tf_graphdef, output_nodes, **kwargs)
File "/usr/lib/python2.7/dist-packages/uff/converters/tensorflow/conversion_helpers.py", line 77, in from_tensorflow
name="main")
File "/usr/lib/python2.7/dist-packages/uff/converters/tensorflow/converter.py", line 74, in convert_tf2uff_graph
uff_graph, input_replacements)
File "/usr/lib/python2.7/dist-packages/uff/converters/tensorflow/converter.py", line 45, in convert_tf2uff_node
tf_node = tf_nodes[name]
KeyError: 'scores boxes classes'
Can you provide script to convert this ssd model to tensorrt plan file?
Hi , I want to convert ssd model to tensorrt plan file . But I am getting error as
Using output node scores boxes classes Converting to UFF graph Traceback (most recent call last): File "convert_plan.py", line 73, in
data_type
File "convert_plan.py", line 24, in frozenToPlan
text=False,
File "/usr/lib/python2.7/dist-packages/uff/converters/tensorflow/conversion_helpers.py", line 113, in from_tensorflow_frozen_model
return from_tensorflow(tf_graphdef, output_nodes, **kwargs)
File "/usr/lib/python2.7/dist-packages/uff/converters/tensorflow/conversion_helpers.py", line 77, in from_tensorflow
name="main")
File "/usr/lib/python2.7/dist-packages/uff/converters/tensorflow/converter.py", line 74, in convert_tf2uff_graph
uff_graph, input_replacements)
File "/usr/lib/python2.7/dist-packages/uff/converters/tensorflow/converter.py", line 45, in convert_tf2uff_node
tf_node = tf_nodes[name]
KeyError: 'scores boxes classes'
Can you provide script to convert this ssd model to tensorrt plan file?
Thanks**