uff

Converting Tensorflow Frozen Graph to UFF for TensorRT inference

被刻印的时光 ゝ 提交于 2020-08-10 05:32:09
问题 I want to use a tensorflow model with TensorRT (C++) and first need to convert the the .pb to .uff format. When executing this code: uff_model = uff.from_tensorflow_frozen_model( frozen_file="my_tf_model.pb", output_nodes=["output_layer"], output_filename="tmp.uff", debug_mode=True, ) I am getting this error message: Traceback (most recent call last): File "/home/jodo/ILONA/object-detection/ssd/src/uff_converter/uff_converter.py", line 29, in <module> text=False, File "/home/jodo/miniconda3