The Transfer Learning Toolkit - Trainer for Intelligent Video Analytics, This Documentation provides:
General overview and capabilities of the Transfer Learning Toolkit.
Instruction on using TLT-Trainer container to start training models with different architectures and backbones.
Explanation about API, code structure, and overall workflow for development and debugging.
This software is used to train computer vision and deep learning models for streaming analytics use cases.
The TLT-Trainer downloads pre-trained weight for each backbone from NGC or can use custom models, each base model is trained on Open Images Dataset. Pre-trained weights provide a great starting point for applying transfer learning on your own dataset.
Currently, the following applications are supported:
Under object detection the following meta architectures are supported:
DSSD YOLOv3 FasterRCNN RetinaNet
Get an NGC API key
Go to NGC and click the Transfer Learning Toolkit container in the Catalog tab. This message is displayed, Sign in to access the PULL feature of this repository.
Enter your email address and click Next or click Create an Account.
Choose your organization when prompted for Organization/Team.
Click Sign In.
Select the Containers tab on the left navigation pane and click the Transfer Learning Toolkit tile.
In the next section, you will learn about architecture specific models and their requirements.