pytorch 加载(.pth)格式的模型实例
有一些非常流行的网络如 resnet、squeezenet、densenet等在pytorch里面都有，包括网络结构和训练好的模型。
pytorch自带模型网址：https://pytorch-cn.readthedocs.io/zh/latest/torchvision/torchvision-models/
按官网加载预训练好的模型：
import torchvision.models as models # pretrained=True就可以使用预训练的模型 resnet18 = models.resnet18(pretrained=True) print(resnet18)
报错如下：
requests.exceptions.ConnectionError: ('Connection aborted.', TimeoutError(10060, '由于连接方在一段时间后没有正确答复或连接的主机没有反应，连接尝试失败。', None, 10060, None))
主要是因为代码会去远端下载模型的参数，而国内的网一般连接不上，这是我们需要手动去下载你要的预训练网络。
通过地址下载，地址有两种获取方式：
1.从报错里面获取，上述代码运行时会出现这样一行信息：
Downloading: "https://download.pytorch.org/models/resnet18-5c106cde.pth" to C:\Users\Luo/.torch\models\resnet18-5c106cde.pth
复制这个网址到浏览器，有可能打不开，去掉https://，直接输入download.pytorch.org/models/resnet18-5c106cde.pth就可以下载了。
2.从pytorch的github下找模型的地址：https://github.com/pytorch/vision/tree/master/torchvision/models
找到对应模型名称点进去找地址
下载好后自行保存，我是直接存在pytorch models里面
接下来就是运行这个.pth文件。首先要判断是保存的整个网络结构加参数呢，还是只保存了参数，可以测试一下。这是我的模型是squeezenet1_1，你可以测试自己下载的模型
import torch pthfile = r'E:\anaconda\app\envs\luo\Lib\site-packages\torchvision\models\squeezenet1_1.pth' net = torch.load(pthfile) print(net)
结果为
很明显就是只保存了参数，这是我们要换个方法加载模型
import torch import torchvision.models as models # pretrained=True就可以使用预训练的模型 net = models.squeezenet1_1(pretrained=False) pthfile = r'E:\anaconda\app\envs\luo\Lib\site-packages\torchvision\models\squeezenet1_1.pth' net.load_state_dict(torch.load(pthfile)) print(net)
结果;
这下就加载好预训练模型了
以上这篇pytorch 加载(.pth)格式的模型实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家详细介绍了matplotlib绘制多子图师如何实现共享坐标轴,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下2024-02-02
这篇文章主要介绍了如何在python字符串中输入纯粹的{}以及python字符串连接的三种方法,需要的朋友可以参考下2018-08-08
可能有人会想到,Python中既然可以直接通过父类名调用父类方法为什么还会存在super函数?其实,很多人对Python中的super函数的认识存在误区,本文我们就带来在Python编程中对super函数的正确理解和用法解析2016-07-07
python实现查找excel里某一列重复数据并且剔除后打印的方法
这篇文章主要介绍了python实现查找excel里某一列重复数据并且剔除后打印的方法,涉及Python使用xlrd模块操作Excel的相关技巧,需要的朋友可以参考下2015-05-05
本文主要介绍了python实现矩阵的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-07-07
这篇文章主要介绍了python基于property()函数定义属性,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-01-01
计算机二级python学习教程（3） python语言基本数据类型
这篇文章主要为大家详细介绍了计算机二级python学习教程的第3篇，python语言基本数据类型，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-05-05
一起解密Python中的*args和**kwargs无限可能的函数参数
这篇文章主要来跟大家一起解密Python中的*args和**kwargs无限可能的函数参数使用的灵活性，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-06-06
今天小编就为大家分享一篇python获取代理IP的实例分享，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
Django Admin后台模型列表页面如何添加自定义操作按钮
这篇文章主要介绍了Django Admin后台模型列表页面如何添加自定义操作按钮,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
最新评论