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