Skip to content

Latest commit

 

History

History
25 lines (18 loc) · 766 Bytes

retinanet.rst

File metadata and controls

25 lines (18 loc) · 766 Bytes

RetinaNet

.. currentmodule:: torchvision.models.detection

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

.. betastatus:: detection module

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.

.. autosummary::
    :toctree: generated/
    :template: function.rst

    retinanet_resnet50_fpn
    retinanet_resnet50_fpn_v2