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

verified publisher icon smistad
→

Metadata

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

← Metadata

18
Stars
2
Forks
Watchers

Owner

verified publisher icon smistad

Metadata

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

Back