Shortcuts

RetinaNet

The RetinaNet model is based on the Focal Loss for Dense Object Detection paper.

Warning

The detection module is in Beta stage, and backward compatibility is not guaranteed.

Model builders

The following model builders can be used to instantiate a RetinaNet model, with or without pre-trained weights. All the model buidlers internally rely on the torchvision.models.detection.retinanet.RetinaNet base class. Please refer to the source code for more details about this class.

retinanet_resnet50_fpn(*[, weights, ...])

Constructs a RetinaNet model with a ResNet-50-FPN backbone.

retinanet_resnet50_fpn_v2(*[, weights, ...])

Constructs an improved RetinaNet model with a ResNet-50-FPN backbone.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources