RetinaNet¶
RetinaNet 模型基於Focal Loss for Dense Object Detection這篇論文。
警告
檢測模組目前處於 Beta 階段,不保證向下相容性。
模型構建器¶
以下模型構建器可用於例項化 RetinaNet 模型,可選擇是否使用預訓練權重。所有模型構建器內部都依賴於torchvision.models.detection.retinanet.RetinaNet基類。有關此類的更多詳細資訊,請參閱原始碼。
|
構建帶有 ResNet-50-FPN 主幹網路的 RetinaNet 模型。 |
|
構建帶有 ResNet-50-FPN 主幹網路的改進版 RetinaNet 模型。 |