Train Adapt Optimize (TAO) Toolkit is a python based AI toolkit for taking purpose-built pre-trained AI models and customizing them with your own data. TAO adapts popular network architectures and backbones to your data, allowing you to train, fine tune, prune and export highly optimized and accurate AI models for edge deployment.
The pre-trained models accelerate the AI training process and reduce costs associated with large scale data collection, labeling, and training models from scratch. Transfer learning with pre-trained models can be used for AI applications in smart cities, retail, healthcare, industrial inspection and more.
Build end-to-end services and solutions for transforming pixels and sensor data to actionable insights using TAO, DeepStream SDK and TensorRT. TAO can train models for common vision AI tasks such as object detection, classification, instance segmentation as well as other complex tasks such as facial landmark, gaze estimation, heart rate estimation and others.
This resource lists out several sample notebooks to walk you through full training workflow using TAO 3.0.
To get started, first choose the model architecture that you want to build, then select the appropriate model card on NGC and then choose one of the supported backbones.
Setup your python environment using python
In TAO Toolkit, we have created an abstraction above the container, you will launch all your training jobs from the launcher. No need to manually pull the appropriate container, tao-launcher will handle that. You may install the launcher using pip with the following commands.
pip3 install nvidia-pyindex pip3 install nvidia-tao
jupyter notebook --ip 0.0.0.0 --port 8888 --allow-root
|Purpose-built Model||Jupyter notebook|
|License Plate Detection||detectnet_v2/detectnet_v2.ipynb|
|License Plate Recognition||lprnet/lprnet.ipynb|
|Heart Rate Estimation||heartratenet/heartratenet.ipynb|
|Open model architecture||Jupyter notebook|