pytorch 加载(.pth)格式的模型实例


Posted in Python onAugust 20, 2019

有一些非常流行的网络如 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 加载(.pth)格式的模型实例

下载好后自行保存,我是直接存在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)

结果为

pytorch 加载(.pth)格式的模型实例

很明显就是只保存了参数,这是我们要换个方法加载模型

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)格式的模型实例

这下就加载好预训练模型了

以上这篇pytorch 加载(.pth)格式的模型实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python文件和目录操作函数小结
Jul 11 Python
python使用分治法实现求解最大值的方法
May 12 Python
python with提前退出遇到的坑与解决方案
Jan 05 Python
Python简单实现查找一个字符串中最长不重复子串的方法
Mar 26 Python
Linux CentOS Python开发环境搭建教程
Nov 28 Python
Python设计模式之抽象工厂模式原理与用法详解
Jan 15 Python
PyQt5显示GIF图片的方法
Jun 17 Python
纯python进行矩阵的相乘运算的方法示例
Jul 17 Python
Django Aggregation聚合使用方法解析
Aug 01 Python
使用python 将图片复制到系统剪贴中
Dec 13 Python
Django密码存储策略分析
Jan 09 Python
keras 读取多标签图像数据方式
Jun 12 Python
python multiprocessing模块用法及原理介绍
Aug 20 #Python
python 并发编程 阻塞IO模型原理解析
Aug 20 #Python
PyTorch中常用的激活函数的方法示例
Aug 20 #Python
Pytorch抽取网络层的Feature Map(Vgg)实例
Aug 20 #Python
python批量解压zip文件的方法
Aug 20 #Python
pytorch获取vgg16-feature层输出的例子
Aug 20 #Python
python 并发编程 非阻塞IO模型原理解析
Aug 20 #Python
You might like
PHP下判断网址是否有效的代码
2011/10/08 PHP
PHP正则表达式笔记与实例详解
2019/05/09 PHP
不使用中间变量,交换int型的 a, b两个变量的值。
2010/10/29 Javascript
jquery 延迟执行实例介绍
2013/08/20 Javascript
Javascript中的apply()方法浅析
2015/03/15 Javascript
js绘制圆形和矩形的方法
2015/08/05 Javascript
JS实现的5级联动Select下拉选择框实例
2015/08/17 Javascript
基于JavaScript实现定时跳转到指定页面
2016/01/01 Javascript
javascript自动切换焦点控制效果完整实例
2016/02/02 Javascript
Bootstrap CDN和本地化环境搭建
2016/10/26 Javascript
Agularjs妙用双向数据绑定实现手风琴效果
2017/05/26 Javascript
详解Vue.js组件可复用性的混合(mixin)方式和自定义指令
2017/09/06 Javascript
浅析Angular 实现一个repeat指令的方法
2019/07/21 Javascript
Vuex,iView UI面包屑导航使用扩展详解
2019/11/04 Javascript
[01:02:46]VGJ.S vs NB 2018国际邀请赛小组赛BO2 第二场 8.18
2018/08/19 DOTA
理解python多线程(python多线程简明教程)
2014/06/09 Python
python实现简易版计算器
2020/06/22 Python
CentOS 7下安装Python3.6 及遇到的问题小结
2018/11/08 Python
python 杀死自身进程的实现方法
2019/07/01 Python
vscode 配置 python3开发环境的方法
2019/09/19 Python
在django项目中导出数据到excel文件并实现下载的功能
2020/03/13 Python
django列表筛选功能的实现代码
2020/03/27 Python
Django实现从数据库中获取到的数据转换为dict
2020/03/27 Python
mac安装python3后使用pip和pip3的区别说明
2020/09/01 Python
python中使用.py配置文件的方法详解
2020/11/23 Python
Python如何使用神经网络进行简单文本分类
2021/02/25 Python
英国最出名高街品牌:Forever Unique
2018/02/24 全球购物
BAILEY 44官网:美国制造的女性服装
2019/07/01 全球购物
职业生涯规划书基本格式
2014/01/06 职场文书
红领巾广播站广播稿
2014/02/01 职场文书
2014年财务工作总结范文
2014/11/11 职场文书
干部个人考察材料
2014/12/24 职场文书
求职自我推荐信
2015/03/24 职场文书
整改通知书格式
2015/04/22 职场文书
公安忠诚教育心得体会
2016/01/23 职场文书
深入讲解数据库中Decimal类型的使用以及实现方法
2022/02/15 MySQL