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