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定时器使用示例分享
Feb 16 Python
Python多线程实例教程
Sep 06 Python
讲解Python中if语句的嵌套用法
May 14 Python
Python正则简单实例分析
Mar 21 Python
基于Django的python验证码(实例讲解)
Oct 23 Python
Python语言生成水仙花数代码示例
Dec 18 Python
Python实现KNN邻近算法
Jan 28 Python
Python cookbook(数据结构与算法)根据字段将记录分组操作示例
Mar 19 Python
Python实现简单查找最长子串功能示例
Feb 26 Python
浅谈Python爬虫基本套路
Mar 25 Python
Python利用神经网络解决非线性回归问题实例详解
Jul 19 Python
python tkinter图形界面代码统计工具
Sep 18 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
无线电波是什么?它是怎样传输的?
2021/03/01 无线电
很好用的PHP数据库类
2009/05/27 PHP
提升PHP性能的21种方法介绍
2013/06/25 PHP
php实现无限级分类
2014/12/24 PHP
PHP4和PHP5版本下解析XML文档的操作方法实例分析
2017/05/20 PHP
论坛里点击别人帖子下面的回复,回复标题变成“回复 24# 的帖子”
2009/06/14 Javascript
jQuery获取注册信息并提示实现代码
2013/04/21 Javascript
js/html光标定位的实现代码
2013/09/23 Javascript
jQuery使用after()方法在元素后面添加多项内容的方法
2015/03/26 Javascript
jQuery控制文本框只能输入数字和字母及使用方法
2016/05/26 Javascript
JavaScript获取URL中参数querystring的方法详解
2016/10/11 Javascript
基于BootstrapValidator的Form表单验证(24)
2016/12/12 Javascript
原生JS京东轮播图代码
2017/03/22 Javascript
AngularJS学习笔记之表单验证功能实例详解
2017/07/06 Javascript
浅谈vue的iview列表table render函数设置DOM属性值的方法
2017/09/30 Javascript
vue中使用cropperjs的方法
2018/03/01 Javascript
Element-ui自定义table表头、修改列标题样式、添加tooltip、:render-header使用
2019/04/11 Javascript
layui内置模块layim发送图片添加加载动画的方法
2019/09/23 Javascript
解决Vue大括号字符换行踩的坑
2020/11/09 Javascript
浅谈Python爬取网页的编码处理
2016/11/04 Python
python 将print输出的内容保存到txt文件中
2018/07/17 Python
python爬虫 爬取58同城上所有城市的租房信息详解
2019/07/30 Python
python通过安装itchat包实现微信自动回复收到的春节祝福
2020/01/19 Python
python等差数列求和公式前 100 项的和实例
2020/02/25 Python
PyTorch在Windows环境搭建的方法步骤
2020/05/12 Python
python 装饰器的使用示例
2020/10/10 Python
H5 canvas实现贪吃蛇小游戏
2017/07/28 HTML / CSS
现代家居用品及礼品:LBC Modern
2018/06/24 全球购物
工程造价专业大学生职业规划范文
2014/03/09 职场文书
大学生两会精神学习心得体会
2014/03/10 职场文书
2015年民主生活会发言材料
2014/12/15 职场文书
初中家长评语和期望
2014/12/26 职场文书
2015年信息中心工作总结
2015/05/25 职场文书
安全生产学习心得体会
2016/01/18 职场文书
九年级数学教学反思
2016/02/17 职场文书
MySQL安装失败的原因及解决步骤
2022/06/14 MySQL