Skip to content

Convert TF/Keras model to TRT engine with ONNX

Notifications You must be signed in to change notification settings

yuanyantai/tf2trt_with_onnx

 
 

Repository files navigation

tf2trt_wtih_onnx

This repo documnet how to convert Tensorflow / Keras model to TRT engine using ONNX.
Note: If tf2onnx didn't work on Keras model try to use keras2onnx

TODOs

When set, TensorFlow fold_constants transformation is applied before conversion. This benefits features including Transpose optimization (e.g. Transpose operations introduced during tf-graph-to-onnx-graph conversion will be removed), and RNN unit conversion (for example LSTM). Older TensorFlow version might run into issues with this option depending on the model.

About

Convert TF/Keras model to TRT engine with ONNX

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 65.0%
  • Python 35.0%