Skip to content
Simple script to convert a frozen tensorflow .pb file to TensorRT UFF format
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
README.md
convert.py

README.md

Setup

Install TensorRT UFF converter:

sudo apt install uff-converter-tf

Usage

python3 convert.py /path/to/your/model.pb name_of_output_node
You can’t perform that action at this time.