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

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.

Hi,

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

Thanks.