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