Dmitry.AI

Nvidia TAO Pretrained DetectNet V2

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

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 DetectNet_v2 object detection networks in Train Adapt Optimize (TAO) Toolkit to facilitate transfer learning.

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

  • resnet10/resnet18/resnet34/resnet50
  • vgg16/vgg19
  • googlenet
  • mobilenet_v1/mobilenet_v2
  • squeezenet
  • darknet19/darknet53

See also: