The TensorFlow models repository offers a streamlined procedure for training image classification and object detection models. In this tutorial we will discuss TensorRT integration in TensorFlow, and how it may be used to accelerate models sourced from the TensorFlow models repository for use on NVIDIA Jetson. We will run through all the steps necessary to train and deploy an object detection model for a custom task.
In this tutorial you will learn:
- How to train an object detection model from the TensorFlow models repository for a custom task
- The performance benefits gained using NVIDIA TensorRT to accelerate image classification and object detection models
- How to optimize a trained object detection model for inference on Jetson