pytorch 更改预训练模型网络结构的方法


Posted in Python onAugust 19, 2019

一个继承nn.module的model它包含一个叫做children()的函数,这个函数可以用来提取出model每一层的网络结构,在此基础上进行修改即可,修改方法如下(去除后两层):

resnet_layer = nn.Sequential(*list(model.children())[:-2])

那么,接下来就可以构建我们的网络了:

class Net(nn.Module):
  def __init__(self , model):
    super(Net, self).__init__()
    #取掉model的后两层
    self.resnet_layer = nn.Sequential(*list(model.children())[:-2])
    
    self.transion_layer = nn.ConvTranspose2d(2048, 2048, kernel_size=14, stride=3)
    self.pool_layer = nn.MaxPool2d(32) 
    self.Linear_layer = nn.Linear(2048, 8)
    
  def forward(self, x):
    x = self.resnet_layer(x)
 
    x = self.transion_layer(x)
 
    x = self.pool_layer(x)
 
    x = x.view(x.size(0), -1) 
 
    x = self.Linear_layer(x)
    
    return x

最后,构建一个对象,并加载resnet预训练的参数就可以啦~

resnet = models.resnet50(pretrained=True)
model = Net(resnet)

以上这篇pytorch 更改预训练模型网络结构的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
在Python中使用swapCase()方法转换大小写的教程
May 20 Python
搭建Python的Django框架环境并建立和运行第一个App的教程
Jul 02 Python
tornado 多进程模式解析
Jan 15 Python
python pandas中DataFrame类型数据操作函数的方法
Apr 08 Python
python 字符串只保留汉字的方法
Nov 16 Python
对PyQt5中的菜单栏和工具栏实例详解
Jun 20 Python
opencv-python 读取图像并转换颜色空间实例
Dec 09 Python
Python导入模块包原理及相关注意事项
Mar 25 Python
python自定义函数def的应用详解
Jun 03 Python
pytorch实现查看当前学习率
Jun 24 Python
学python需要去培训机构吗
Jul 01 Python
Python QT组件库qtwidgets的使用
Nov 02 Python
pytorch打印网络结构的实例
Aug 19 #Python
pytorch索引查找 index_select的例子
Aug 18 #Python
浅谈Pytorch中的torch.gather函数的含义
Aug 18 #Python
PyTorch中Tensor的维度变换实现
Aug 18 #Python
PyTorch中Tensor的拼接与拆分的实现
Aug 18 #Python
详解PyTorch中Tensor的高阶操作
Aug 18 #Python
浅析PyTorch中nn.Linear的使用
Aug 18 #Python
You might like
PHP 5昨天隆重推出--PHP 5/Zend Engine 2.0新特性
2006/10/09 PHP
php缩放图片(根据宽高的等比例缩放)实例介绍
2013/06/09 PHP
thinkPHP查询方式小结
2016/01/09 PHP
php封装db类连接sqlite3数据库的方法实例
2017/12/19 PHP
thinkPHP和onethink微信支付插件分享
2019/08/11 PHP
JavaScript中的对象化编程
2008/01/16 Javascript
ExtJs中简单的登录界面制作方法
2010/08/19 Javascript
自定义ExtJS控件之下拉树和下拉表格附源码
2013/10/15 Javascript
JavaScript实现拖拽网页内元素的方法
2015/04/15 Javascript
jquery实现动静态条形统计图
2015/08/17 Javascript
javascript伸缩型菜单实现代码
2015/11/16 Javascript
基于JavaScript实现Json数据根据某个字段进行排序
2015/11/24 Javascript
vue组件中点击按钮后修改输入框的状态实例代码
2017/04/14 Javascript
ES6新特性五:Set与Map的数据结构实例分析
2017/04/21 Javascript
Dropify.js图片宽高自适应的方法
2017/11/27 Javascript
关于vue-cli 3配置打包优化要点(推荐)
2019/04/22 Javascript
Vue实现push数组并删除的例子
2019/11/01 Javascript
html中创建并调用vue组件的几种方法汇总
2020/11/17 Javascript
vue3.0中友好使用antdv示例详解
2021/01/05 Vue.js
[01:57]DOTA2上海特锦赛小组赛解说单车采访花絮
2016/02/27 DOTA
[01:32]完美世界DOTA2联赛10月29日精彩集锦
2020/10/30 DOTA
Python中用pycurl监控http响应时间脚本分享
2015/02/02 Python
python使用urlparse分析网址中域名的方法
2015/04/15 Python
编写Python爬虫抓取豆瓣电影TOP100及用户头像的方法
2016/01/20 Python
浅谈django model的get和filter方法的区别(必看篇)
2017/05/23 Python
django中SMTP发送邮件配置详解
2019/07/19 Python
Python如何把十进制数转换成ip地址
2020/05/25 Python
Python中bisect的用法及示例详解
2020/07/20 Python
css3新单位vw、vh的使用教程
2018/03/23 HTML / CSS
澳大利亚第一的设计师礼服租赁网站:GlamCorner
2017/08/13 全球购物
一份全面的PHP面试问题考卷
2012/07/15 面试题
党员十八大心得体会
2014/09/12 职场文书
2014年庆祝国庆65周年演讲稿
2014/09/21 职场文书
2014年作风建设工作总结
2014/10/29 职场文书
保护地球的宣传语
2015/07/13 职场文书
清明节主题班会
2015/08/14 职场文书