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 变量类型及命名规则介绍
Jun 08 Python
python使用urllib模块开发的多线程豆瓣小站mp3下载器
Jan 16 Python
Python help()函数用法详解
Mar 11 Python
Python中文编码那些事
Jun 25 Python
Python实现的文本编辑器功能示例
Jun 30 Python
python3利用tcp实现文件夹远程传输
Jul 28 Python
pycham查看程序执行的时间方法
Nov 29 Python
Python I/O与进程的详细讲解
Mar 08 Python
python模块之subprocess模块级方法的使用
Mar 26 Python
django实现类似触发器的功能
Nov 15 Python
Python matplotlib实时画图案例
Apr 23 Python
pytorch model.cuda()花费时间很长的解决
Jun 01 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
基于mysql的论坛(2)
2006/10/09 PHP
隐藏你的.php文件的实现方法
2007/03/19 PHP
精通php的十大要点(上)
2009/02/04 PHP
初识php MVC
2014/09/10 PHP
PHP中的Streams详细介绍
2014/11/12 PHP
如何使用jQuery+PHP+MySQL来实现一个在线测试项目
2015/04/26 PHP
[原创]CI(CodeIgniter)简单统计访问人数实现方法
2016/01/19 PHP
js 数值项目的格式化函数代码
2010/05/14 Javascript
jquery自动将form表单封装成json的具体实现
2014/03/17 Javascript
Javascript实现苹果悬浮虚拟按钮
2016/04/10 Javascript
Angular懒加载机制刷新后无法回退的快速解决方法
2016/08/30 Javascript
JQuery 动态生成Table表格实例代码
2016/12/02 Javascript
Ajax高级笔记 JavaScript高级程序设计笔记
2017/06/22 Javascript
基于JavaScript实现微信抢红包功能
2017/07/20 Javascript
jQuery分组选择器简单用法示例
2019/04/04 jQuery
我要点爆”微信小程序云开发之项目建立与我的页面功能实现
2019/05/26 Javascript
在Vue中使用icon 字体图标的方法
2019/06/14 Javascript
Vue分页插件的前后端配置与使用
2019/10/09 Javascript
JS数组方法concat()用法实例分析
2020/01/18 Javascript
解决Vue 移动端点击出现300毫秒延迟的问题
2020/07/21 Javascript
[03:49]辉夜杯现场龙骑士COSER秀情商“我喜欢芬队!”
2015/12/27 DOTA
python3+PyQt5 使用三种不同的简便项窗口部件显示数据的方法
2019/06/17 Python
Pyqt QImage 与 np array 转换方法
2019/06/27 Python
Django模型修改及数据迁移实现解析
2019/08/01 Python
Python Opencv提取图片中某种颜色组成的图形的方法
2019/09/19 Python
Python reshape的用法及多个二维数组合并为三维数组的实例
2020/02/07 Python
Django 博客实现简单的全文搜索的示例代码
2020/02/17 Python
巴西独家产品和现场演示购物网站:Shoptime
2019/07/11 全球购物
ABOUT YOU匈牙利:500个最受欢迎的时尚品牌
2019/07/19 全球购物
生产部管理制度
2014/01/31 职场文书
太太口服液广告词
2014/03/20 职场文书
2014最新开业庆典策划方案(5篇)
2014/09/15 职场文书
群众路线查摆问题及整改措施
2014/10/10 职场文书
房屋产权证明书
2014/10/15 职场文书
2016党员三严三实心得体会
2016/01/15 职场文书
python中sys模块的介绍与实例
2021/04/17 Python