RetinaNet

RetinaNet 模型基於 Focal Loss for Dense Object Detection 論文。

警告

檢測模組處於 Beta 階段，不保證向後相容。

模型構建器

以下模型構建器可用於例項化 RetinaNet 模型，可帶或不帶預訓練權重。所有模型構建器內部都依賴於 torchvision.models.detection.retinanet.RetinaNet 基類。有關此類的更多詳細資訊，請參閱 原始碼。

retinanet_resnet50_fpn(*[, weights, ...])	構建一個帶有 ResNet-50-FPN 主幹的 RetinaNet 模型。
retinanet_resnet50_fpn_v2(*[, weights, ...])	構建一個改進的 RetinaNet 模型，帶有 ResNet-50-FPN 主幹。