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


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 builders 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.


Access comprehensive developer documentation for PyTorch

View Docs


Get in-depth tutorials for beginners and advanced developers

View Tutorials


Find development resources and get your questions answered

View Resources