NVIDIA-AI-IOT / tf_trt_models

TensorFlow models accelerated with NVIDIA TensorRT
BSD 3-Clause "New" or "Revised" License
682 stars 245 forks source link

regarding conversion of ssd_inception_v2_coco_trt.pb to tensorrt plan #4

Open jumpstart8 opened 6 years ago

jumpstart8 commented 6 years ago

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**