PyTorch 1.0.0图像预训练模型

文件列表(压缩包大小 269.99M)      点击页面右上角“下载”按钮下载

概述

介绍 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

理工酷提示:

如果遇到文件不能下载或其他产品问题,请添加管理员微信:ligongku666,并备注:产品反馈

评论(0)

0/250
免费
赞 1
收藏 0
评论 0
举报

网站声明:本站所有资源均为用户上传,如果侵犯了您的合法权益,请点击上方举报按钮,或添加管理员微信:ligongku666 ,并备注:举报。我们将快速核实并处理。

文件编号:397
上传时间:2021-01-22
文件大小:269.99M

阿托

声望 • 7052

分类:
计算机/软件
标签:
算法人工智能