Skip to content

Latest commit

 

History

History

tf1

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 

Code related to model quantisation with TensorRT for TensorFlow 1.x

1. your_model

Well... as the title suggests Creates a MobileNetV2 model instance and saves it as a Keras .h5 file

2. freeze_build_trt

Freezes the previously created model, optimises it with TensorRT and saves the optimised model engine

3. load_trt

Loads optimised TensorRT model engine and tests it

model_data (generated)

Text file which contains model input and output layer info