We’re happy to share the following project on GitHub which demonstrates object detection and image classification workflows using TensorRT integration in TensorFlow (for details on TF-TRT integration see this blog post). With this project you can easily accelerate popular models like SSD Inception V2 for use on Jetson.
The project is hosted at the following URL
By following the steps outlined in this project you will
- Download pretrained object detection and image classification models sourced from the TensorFlow models repository
- Run scripts to preprocess the TensorFlow graphs for best utilization of TensorRT and Jetson
- Accelerate models using TensorRT integration in TensorFlow
- Execute models with the TensorFlow Python API
The models are sourced from the TensorFlow models repository, so it is possible to train the models for custom tasks using the steps detailed there. Provided you use one of the listed model architectures, you can follow the steps above to easily accelerate the model for ideal performance on Jetson.