就拿Resnet18舉例
在程序中輸入
from __future__ import print_function, division
from torchvision import models
model_ft = models.resnet18(pretrained=True)
- 1
- 2
- 3
- 4
然後運行，就會出現如下內容
再將這個網址複製到瀏覽器中，就可以直接下載Resnet18模型。下載結束後，將下載的文件放入你建立的python項目中，再運用如下代碼就可以調用Resnet18模型。
model_ft = models.resnet18(pretrained=False)
pre = torch.load('resnet18-5c106cde.pth')
model_ft.load_state_dict(pre)
- 1
- 2
- 3
希望對你有幫助！
其他的學習模型也可以用類似的方法下載，或者可以參看下面的下載網址。
‘resnet18’: ‘https://download.pytorch.org/models/resnet18-5c106cde.pth’
‘resnet34’: ‘https://download.pytorch.org/models/resnet34-333f7ec4.pth’
‘resnet50’: ‘https://download.pytorch.org/models/resnet50-19c8e357.pth’
‘resnet101’: ‘https://download.pytorch.org/models/resnet101-5d3b4d8f.pth’
‘resnet152’: ‘https://download.pytorch.org/models/resnet152-b121ed2d.pth’
‘densenet121’: ‘https://download.pytorch.org/models/densenet121-a639ec97.pth’
‘densenet169’: ‘https://download.pytorch.org/models/densenet169-b2777c0a.pth’
‘densenet201’: ‘https://download.pytorch.org/models/densenet201-c1103571.pth’
‘densenet161’: ‘https://download.pytorch.org/models/densenet161-8d451a50.pth’
‘vgg11’: ‘https://download.pytorch.org/models/vgg11-bbd30ac9.pth’
‘vgg13’: ‘https://download.pytorch.org/models/vgg13-c768596a.pth’
‘vgg16’: ‘https://download.pytorch.org/models/vgg16-397923af.pth’
‘vgg19’: ‘https://download.pytorch.org/models/vgg19-dcbb9e9d.pth’
‘vgg11_bn’: ‘https://download.pytorch.org/models/vgg11_bn-6002323d.pth’
‘vgg13_bn’: ‘https://download.pytorch.org/models/vgg13_bn-abd245e5.pth’
‘vgg16_bn’: ‘https://download.pytorch.org/models/vgg16_bn-6c64b313.pth’
‘vgg19_bn’: ‘https://download.pytorch.org/models/vgg19_bn-c79401a0.pth’
‘alexnet’: ‘https://download.pytorch.org/models/alexnet-owt-4df8aa71.pth’
#Inception v3 ported from TensorFlow
‘inception_v3_google’: ‘https://download.pytorch.org/models/inception_v3_google-1a9a5a14.pth’
</div>
<link href="https://csdnimg.cn/release/phoenix/mdeditor/markdown_views-60ecaf1f42.css" rel="stylesheet">
</div>