Regarding training custom models with TensorFlow and putting them to use on Jetson Nano for inference

Autonomous Machines Jetson & Embedded Systems Jetson Nano
1

Hi,

I am looking for good resources which guide how one can train a custom model using TensorFlow (and Keras) and use TensorRT to further optimize them so that they could be used on Jetson Nano for inference.

Thank you in advance.

2

Hi,

You can check the tutorial below based on the different use case.
Classification: [url]https://github.com/NVIDIA-AI-IOT/tf_to_trt_image_classification[/url]
Detection: [url]https://github.com/NVIDIA-AI-IOT/tf_trt_models[/url]

Thanks.