NVIDIA TAO (Object detection & Images)

The NVIDIA TAO Toolkit built on TensorFlow and PyTorch, uses the power of transfer learning while simultaneously simplifying the model training process and optimizing the model for inference throughput on the target platform. The result is an ultra-streamlined workflow. Take your own models or pre-trained models, adapt them to your own real or synthetic data, then optimize for inference throughput. All without needing AI expertise or large training datasets.

Edge Impulse offers the following learning blocks for NVIDIA TAO object detection and image classification tasks: RetinaNet, YOLOv3, YOLOv4, SSD, and image classification.

Only available for enterprise customers

As this learning block uses GPU hours for training, this integration is only available for enterprise customers. View our pricing for more information.

How to get started?

To build your first object detection models using MobileNetV2 SSD FPN-Lite:

  1. Create a new project in Edge Impulse.

  2. Make sure to set your labelling method to 'Bounding boxes (object detection)' or 'One label per data item (image classification)'.

  3. Collect and prepare your dataset as in object detection or image classification.

  4. Resize your images.

  5. Add an 'NVIDIA TAO ...' block to your impulse.

  1. Under NVIDIA TAO..., select between various parameters, in total there are 88 object detection architectures, and 15 image classification architectures.

    • There are pre-trained 3x224x224 backbones from the NVIDIA TAO catalog, and others trained by Edge Impulse on ImageNet.

    • For image classification, pre-trained weights only support 224x224 image resolution. Image width and height must be greater than 32.

  2. Click on 'Start training'

Next steps: building a machine learning model

With everything setup you can now build your machine learning model with these tutorials:

Last updated