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算法学习之计数排序实例
Dec 18 Python
python获取文件真实链接的方法,针对于302返回码
May 14 Python
Python实现基于C/S架构的聊天室功能详解
Jul 07 Python
Django框架中间件(Middleware)用法实例分析
May 24 Python
Python+pyplot绘制带文本标注的柱状图方法
Jul 08 Python
Python爬取破解无线网络wifi密码过程解析
Sep 17 Python
Python matplotlib以日期为x轴作图代码实例
Nov 22 Python
如何使用python实现模拟鼠标点击
Jan 06 Python
python的help函数如何使用
Jun 11 Python
详解Python中@staticmethod和@classmethod区别及使用示例代码
Dec 14 Python
Python实现淘宝秒杀功能的示例代码
Jan 19 Python
K近邻法(KNN)相关知识总结以及如何用python实现
Jan 28 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
Zend Guard一些常见问题解答
2008/09/11 PHP
php防攻击代码升级版
2010/12/29 PHP
shopex主机报错误请求解决方案(No such file or directory)
2011/12/27 PHP
十幅图告诉你什么是PHP引用
2015/02/22 PHP
php使用strip_tags()去除html标签仍有空白的解决方法
2016/07/28 PHP
smarty模板数学运算示例
2016/12/11 PHP
php 如何设置一个严格控制过期时间的session
2017/05/05 PHP
Thinkphp5.0框架视图view的循环标签用法示例
2019/10/12 PHP
Javascript学习笔记9 prototype封装继承
2010/01/11 Javascript
一段批量给页面上的控件赋值js
2010/06/19 Javascript
面向对象设计模式的核心法则
2013/11/10 Javascript
BootStrap的table表头固定tbody滚动的实例代码
2016/08/24 Javascript
AngularJs bootstrap详解及示例代码
2016/09/01 Javascript
JS中微信小程序自定义底部弹出框
2016/12/22 Javascript
Bootstrap 下拉多选框插件Bootstrap Multiselect
2017/01/22 Javascript
jQuery实现三级联动效果
2017/03/02 Javascript
jquery学习笔记之无new构建详解
2017/12/07 jQuery
Angular ElementRef简介及其使用
2018/10/01 Javascript
Vant的安装和配合引入Vue.js项目里的方法步骤
2018/12/05 Javascript
通过vue手动封装on、emit、off的代码详解
2019/05/29 Javascript
js实现鼠标拖拽div左右滑动
2020/01/15 Javascript
python中map、any、all函数用法分析
2015/04/21 Python
Django中模版的子目录与include标签的使用方法
2015/07/16 Python
python3 读写文件换行符的方法
2018/04/09 Python
python使用magic模块进行文件类型识别方法
2018/12/08 Python
Python Matplotlib库安装与基本作图示例
2019/01/09 Python
Django csrf 两种方法设置form的实例
2019/02/03 Python
使用celery执行Django串行异步任务的方法步骤
2019/06/06 Python
波兰最大的度假胜地和城市公寓租赁运营商:Sun & Snow
2018/10/18 全球购物
纽约海:Sea New York
2018/11/04 全球购物
创意活动策划书
2014/01/15 职场文书
校园文明标语
2014/06/13 职场文书
大学生个人学习总结
2015/02/15 职场文书
离婚起诉书范文2016
2015/11/26 职场文书
干货:如何写好工作总结报告!
2019/05/10 职场文书
html输入两个数实现加减乘除功能
2021/07/01 HTML / CSS