convert-tensorflow-model-to-tensorrt-uff icon indicating copy to clipboard operation
convert-tensorflow-model-to-tensorrt-uff copied to clipboard

Simple script to convert a frozen tensorflow .pb file to TensorRT UFF format

Setup

Install TensorRT UFF converter:

sudo apt install uff-converter-tf

Usage

python3 convert.py /path/to/your/model.pb name_of_output_node