Converting from TensorFlow 1 to TensorRT

Is there a simple way of converting a frozen graph model from TensorFlow 1 to a .UFF file for use with TensorRT? I have done some research into this but haven’t found anything that’s easy to do - is there an easy tutorial out there? Or is there a python script which I can just use so that I can convert from TF1 to TRT.

Thanks in advance!

Hi,

You can find an example below:

In general, you need to run a similar command as below to convert a .pb file into .uff.

$ convert-to-uff -p config.py -O dense_class/Softmax -O dense_regress/BiasAdd -O proposal faster_rcnn.pb

Thanks.

I tried to do this but just got a bunch of errors when I tried to convert.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.