Detectnet_v2 (NVIDIA example)
The TAO Toolkit incorporates Jupyter Notebook to efficiently run and execute tasks. Every step from environment setup to model deployment can be done using the Jupyter Notebook
The following project follows:
pre-requisites
preparing the dataset and pre-trained model
Providing training specification
Running the TAO Training
Evaluating the trained model
Pruning the trained model
Re-training the pruned model
Evaluating the trained model
Visualizing the inferences
Exporting the model
Verifying the deployed model