Dmitry.AI

Nvidia TAO Pretrained Object Detection

catalog.ngc.nvidia.com/orgs/nvidia/teams/tao/models/pretrained_object_detection

Object detection is a popular computer vision technique that can detect one or multiple objects in a frame.
Object detection will recognize the individual objects in an image and places bounding boxes around the object.
This model card contains pretrained weights that may be used as a starting point with the following object detection networks in Train Adapt Optimize (TAO) Toolkit to facilitate transfer learning:

  • YOLOV3
  • FasterRCNN
  • SSD
  • DSSD
  • RetinaNet

It is trained on a subset of the Google OpenImages dataset.
Following backbones are supported with these detection networks.

  • resnet10/resnet18/resnet34/resnet50/resnet101
  • vgg16/vgg19
  • googlenet
  • mobilenet_v1/mobilenet_v2
  • squeezenet
  • darknet19/darknet53
  • efficientnet_b0
  • cspdarknet19/cspdarknet53

See also: