ResNet¶
ResNet 模型基於 《用於影像識別的深度殘差學習》 論文。
注意
TorchVision 的瓶頸將下采樣步幅放在第二個 3x3 卷積上,而原始論文將其放在第一個 1x1 卷積上。這個變體提高了準確性,並被稱為 ResNet V1.5。
模型構建器¶
可以使用以下模型構建器來例項化 ResNet 模型,可選擇是否載入預訓練權重。所有模型構建器內部都依賴於 torchvision.models.resnet.ResNet 基類。有關此類的更多詳細資訊,請參閱 原始碼。
|
ResNet-18,來自 《用於影像識別的深度殘差學習》。 |
|
ResNet-34,來自 《用於影像識別的深度殘差學習》。 |
|
ResNet-50,來自 《用於影像識別的深度殘差學習》。 |
|
ResNet-101,來自 《用於影像識別的深度殘差學習》。 |
|
ResNet-152,來自 《用於影像識別的深度殘差學習》。 |