Tensorflow to TensorRT Model Conversion for Retraining the Model

Autonomous Machines Jetson & Embedded Systems Jetson Xavier NX
1

Hi,

We use Jetson-inference’s pre-trained object detection model for our application. But we want to train weights with our site dataset. Would you provide a source for converting a Tensorflow model to TensorRT for Xavier Nx?

Thank you

3

Hi,

Please convert your model into onnx format with tf2onnx or keras2onnx, and following this sample for TensorRT:

Thanks.

5

@myagmur did you get a performance boost on the nx when you converted your tf model to a tf_rt model?

6

you have to convert your model into TensorRT to run it on NX

7

I did, I am using the process described here: Accelerating Inference In TF-TRT User Guide :: NVIDIA Deep Learning Frameworks Documentation

I am not seeing any performance improvement.