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, ...])	建構具有改良 ResNet-50-FPN 主幹網路的 RetinaNet 模型。