torch.hub.load(‘pytorch/vision‘, ‘deeplabv3_resnet101‘, pretrained=True)运行方法

首先下载 vision-master.zip

 torch_1.7.1_gpu) E:\PythonPRO\PytorchPro\G5>pip install vision-master.zip

实现上安装到 C:\Users\Administrator.cache\torch\hub (每个人安装的地方不一样，通过运行程序可以看出在什么地方)

下面主要用到2个文件

D:\ProgramData\Anaconda3\envs\pytorch_gpu\Lib\site-packages\torch\hub.py
C:\Users\Administrator\.cache\torch\hub\pytorch_vision_master\hubconf.py

在hubconf.py中，并按着下面注释掉不要的：

# Optional list of dependencies required by the package
dependencies = ['torch']# classification
from torchvision.models.alexnet import alexnet
from torchvision.models.densenet import densenet121, densenet169, densenet201, densenet161
from torchvision.models.inception import inception_v3
from torchvision.models.resnet import resnet18, resnet34, resnet50, resnet101, resnet152,\resnext50_32x4d, resnext101_32x8d, wide_resnet50_2, wide_resnet101_2
from torchvision.models.squeezenet import squeezenet1_0, squeezenet1_1
from torchvision.models.vgg import vgg11, vgg13, vgg16, vgg19, vgg11_bn, vgg13_bn, vgg16_bn, vgg19_bn
from torchvision.models.googlenet import googlenet
from torchvision.models.shufflenetv2 import shufflenet_v2_x0_5, shufflenet_v2_x1_0
# from torchvision.models.mobilenetv2 import mobilenet_v2
# from torchvision.models.mobilenetv2 import mobilenetv2
# from torchvision.models.mobilenetv3 import mobilenet_v3_large, mobilenet_v3_small
# from torchvision.models.mnasnet import mnasnet0_5, mnasnet0_75, mnasnet1_0, \
#     mnasnet1_3# segmentation
# from torchvision.models.segmentation import fcn_resnet50, fcn_resnet101, \
#     deeplabv3_resnet50, deeplabv3_resnet101, deeplabv3_mobilenet_v3_large, lraspp_mobilenet_v3_largefrom torchvision.models.segmentation import fcn_resnet50, fcn_resnet101, \deeplabv3_resnet50, deeplabv3_resnet101

根据网址：
https://download.pytorch.org/models/resnet101-5d3b4d8f.pth 下载后放到：
C:\Users\Administrator.cache\torch\hub\checkpoints
https://download.pytorch.org/models/deeplabv3_resnet101_coco-586e9e4e.pth

下载后放到
C:\Users\Administrator.cache\torch\hub\checkpoints
再次运行即可成功。

另外 1_seg_Demo.py文件内容也需要改
将

model = torch.hub.load('pytorch/vision', 'deeplabv3plus_resnet101', pretrained=True) # pretrained=True 设为预训练模式

改为：

model = torch.hub.load('pytorch/vision', 'deeplabv3_resnet101', pretrained=True)  # pretrained=True 设为预训练模式

torch.hub.load(‘pytorch/vision‘, ‘deeplabv3_resnet101‘, pretrained=True)运行方法相关推荐

【pytorch】——torch.hub
[pytorch]--torch.hub Facebook官方博客表示,PyTorch Hub是一个简易API和工作流程,为复现研究提供了基本构建模块,包含预训练模型库.并且,PyTorch Hub还 ...
PyTorch 源码解读之 torch.serialization torch.hub
作者 | 123456 来源 | OpenMMLab 编辑 | 极市平台导读本文解读基于PyTorch 1.7版本,对torch.serialization.torch.save和torch.hu ...
1 | 给马穿上斑马皮的生成对抗网络小实验以及Torch Hub简介
GAN网络实验今天这一节还是练手小实验. 这个实验的方案是使用一个能把马变成斑马的网络,这个网络是基于GAN(generative adversarial network 生成对抗网络)来构建的.所 ...
PyTorch 1.0 中文文档：torch.hub
译者:kunwuz torch.hub.load(github, model, force_reload=False, *args, **kwargs) 从github上加载一个带有预训练权重的模型. ...
pytorch版本问题RuntimeError: params/unet.pth is a zip archive (did you mean to use torch.jit.load()?)
pytorch版本问题RuntimeError: params/unet.pth is a zip archive (did you mean to use torch.jit.load()?) 问题 ...
.pth is a zip archive (did you mean to use torch.jit.load()?)
RuntimeError: bert.pth is a zip archive (did you mean to use torch.jit.load()?) .pth is a zip archiv ...
使用torchvision.models.inception_v3(pretrained=True)加载预训练的模型每次都特别慢
欢迎大家关注笔者,你的关注是我持续更博的最大动力原创文章,转载告知,盗版必究使用torchvision.models.inception_v3(pretrained=True)加载预训练的模型每次 ...
RuntimeError: xxx.pth is a zip archive (did you mean to use torch.jit.load()?)
用高版本的pytorch训练模型,在低版本中测试时出现如下错误: RuntimeError: all_ep_296.pth is a zip archive (did you mean to use ...
xxx.pth或者xxx.pt is a zip archive(did you mean to use torch.jit.load()?问题解决
提示:推荐使用方法三,不会"科学上网"的请关注微信公众号"DGX杂学"并在后台私信! 错误:使用torch.load()加载模型参数时,提示"xxx. ...

torch.hub.load(‘pytorch/vision‘, ‘deeplabv3_resnet101‘, pretrained=True)运行方法

torch.hub.load(‘pytorch/vision‘, ‘deeplabv3_resnet101‘, pretrained=True)运行方法相关推荐

最新文章

热门文章