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.
|
Constructs a RetinaNet model with a ResNet-50-FPN backbone. |
|
Constructs an improved RetinaNet model with a ResNet-50-FPN backbone. |
