理工酷

介绍 PyTorch 1.0.0的Densenet和Resnet预训练模型。

共包含4个文件：

• densenet121.pth
• densenet201.pth
• resnet34.pth
• resnet50.pth

用法

import torch
from torchvision import models, transforms

# densenet121
model = models.densenet121()
model.load_state_dict(torch.load('densenet121.pth'))

# densenet201
model = models.densenet201()
model.load_state_dict(torch.load('densenet201.pth'))

# resnet50
model = models.resnet50()
model.load_state_dict(torch.load('resnet50.pth'))

# resnet34
model = models.resnet34()
model.load_state_dict(torch.load('resnet34.pth'))

转载https://www.kaggle.com/bminixhofer/pytorch-pretrained-image-models?select=resnet50.pth