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解析模块(ConfigParser)使用方法
Dec 10 Python
Python实现的简单hangman游戏实例
Jun 28 Python
python函数式编程学习之yield表达式形式详解
Mar 25 Python
Python常用字符串替换函数strip、replace及sub用法示例
May 21 Python
python实现俄罗斯方块
Jun 26 Python
python的set处理二维数组转一维数组的方法示例
May 31 Python
利用Python进行图像的加法,图像混合(附代码)
Jul 14 Python
Python csv模块使用方法代码实例
Aug 29 Python
python字符串,元组,列表,字典互转代码实例详解
Feb 14 Python
python实现图像拼接
Mar 05 Python
Python通过fnmatch模块实现文件名匹配
Sep 30 Python
python 实现Requests发送带cookies的请求
Feb 08 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
phpMyAdmin 安装配置方法和问题解决
2009/06/08 PHP
php中url函数介绍及使用示例
2014/02/13 PHP
在Thinkphp中使用ajax实现无刷新分页的方法
2016/10/25 PHP
Javascript的常规数组和关联数组对比小结
2012/05/24 Javascript
jquery.form.js用法之清空form的方法
2014/03/07 Javascript
jquery中ajax使用error调试错误的方法
2015/02/08 Javascript
jquery对象访问是什么及使用方法介绍
2016/05/03 Javascript
Node.js 文件夹目录结构创建实例代码
2016/07/08 Javascript
JavaScript跨域调用基于JSON的RESTful API
2016/07/09 Javascript
Vuejs第八篇之Vuejs组件的定义实例解析
2016/09/05 Javascript
angular+ionic 的app上拉加载更新数据实现方法
2017/01/16 Javascript
学好js,这些js函数概念一定要知道【推荐】
2017/01/19 Javascript
使用bootstrap-paginator.js 分页来进行ajax 异步分页请求示例
2017/03/09 Javascript
详解nodejs微信公众号开发——3.封装消息响应模块
2017/04/10 NodeJs
原生js jquery ajax请求以及jsonp的调用方法
2017/08/04 jQuery
Vue数组更新及过滤排序功能
2017/08/10 Javascript
简单了解TypeScript中如何继承 Error 类
2019/06/21 Javascript
解决ele ui 表格表头太长问题的实现
2019/11/13 Javascript
jQuery模仿ToDoList实现简单的待办事项列表
2019/12/30 jQuery
Python正规则表达式学习指南
2016/08/02 Python
Python3.6.0+opencv3.3.0人脸检测示例
2018/05/25 Python
Tensorflow 实现修改张量特定元素的值方法
2018/07/30 Python
Django 视图层(view)的使用
2018/11/09 Python
如何使用Python实现斐波那契数列
2019/07/02 Python
Python拼接字符串的7种方式详解
2020/03/19 Python
python seaborn heatmap可视化相关性矩阵实例
2020/06/03 Python
John Varvatos官方网站:设计师男士时装
2017/02/08 全球购物
HTC VIVE美国官网:VR虚拟现实眼镜
2018/02/13 全球购物
英文版餐饮业求职信
2013/10/18 职场文书
学校采购员岗位职责
2014/01/02 职场文书
家长学校工作方案
2014/05/07 职场文书
2014年前台文员工作总结
2014/12/08 职场文书
2015年路政工作总结
2015/05/22 职场文书
2016廉洁从业学习心得体会
2016/01/19 职场文书
教你漂亮打印Pandas DataFrames和Series
2021/05/29 Python
oracle连接ODBC sqlserver数据源的详细步骤
2021/07/25 Oracle