pytorch 修改预训练model实例


Posted in Python onJanuary 18, 2020

我就废话不多说了,直接上代码吧!

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 = models.resnet50(pretrained=True)
model = Net(resnet)

以上这篇pytorch 修改预训练model实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python 自动提交和抓取网页
Jul 13 Python
python 生成不重复的随机数的代码
May 15 Python
Python中序列的修改、散列与切片详解
Aug 27 Python
Python 逐行分割大txt文件的方法
Oct 10 Python
对python产生随机的二维数组实例详解
Dec 13 Python
pyqt5对用qt designer设计的窗体实现弹出子窗口的示例
Jun 19 Python
Python登录系统界面实现详解
Jun 25 Python
jupyter notebook 重装教程
Apr 16 Python
Python基于BeautifulSoup爬取京东商品信息
Jun 01 Python
python使用openpyxl库读写Excel表格的方法(增删改查操作)
May 02 Python
Python道路车道线检测的实现
Jun 27 Python
Python中字符串对象语法分享
Feb 24 Python
Pytorch自己加载单通道图片用作数据集训练的实例
Jan 18 #Python
pyinstaller 3.6版本通过pip安装失败的解决办法(推荐)
Jan 18 #Python
Python实现点云投影到平面显示
Jan 18 #Python
Pytorch 实现计算分类器准确率(总分类及子分类)
Jan 18 #Python
在pytorch 中计算精度、回归率、F1 score等指标的实例
Jan 18 #Python
Python中实现输入超时及如何通过变量获取变量名
Jan 18 #Python
Pytorch 计算误判率,计算准确率,计算召回率的例子
Jan 18 #Python
You might like
索尼SONY ICF-SW7600GR电路分析与改良
2021/03/02 无线电
php的array_multisort()使用方法介绍
2012/05/16 PHP
php单例模式的简单实现方法
2016/06/10 PHP
详解PHP中foreach的用法和实例
2016/10/25 PHP
Mac系统完美安装PHP7详细教程
2017/06/06 PHP
类之Prototype.js学习
2007/06/13 Javascript
JS简单实现登陆验证附效果图
2013/11/19 Javascript
js使用eval解析json实例与注意事项分享
2014/01/18 Javascript
jquery使用ajax实现微信自动回复插件
2014/04/28 Javascript
node.js中的fs.chmodSync方法使用说明
2014/12/18 Javascript
JavaScript验证Email(3种方法)
2015/09/21 Javascript
跟我学习javascript的闭包
2015/11/16 Javascript
jquery实现ajax加载超时提示的方法
2016/07/23 Javascript
JavaScript中的对象继承关系
2016/08/01 Javascript
js 自带的sort() 方法全面了解
2016/08/16 Javascript
原生js实现电商侧边导航效果
2017/01/19 Javascript
layui 实现table翻页滚动条位置保持不变的例子
2019/09/05 Javascript
详解Vue的七种传值方式
2021/02/08 Vue.js
python操作数据库之sqlite3打开数据库、删除、修改示例
2014/03/13 Python
python django集成cas验证系统
2014/07/14 Python
python魔法方法-属性访问控制详解
2016/07/25 Python
Python os.rename() 重命名目录和文件的示例
2018/10/25 Python
OpenCV HSV颜色识别及HSV基本颜色分量范围
2019/03/22 Python
opencv 图像滤波(均值,方框,高斯,中值)
2020/07/08 Python
python为什么要安装到c盘
2020/07/20 Python
如何利用pycharm进行代码更新比较
2020/11/04 Python
pycharm 复制代码出现空格的解决方式
2021/01/15 Python
基于HTML5 Canvas:字符串,路径,背景,图片的详解
2013/05/09 HTML / CSS
植村秀加拿大官网:Shu Uemura加拿大
2019/09/03 全球购物
保加利亚手表、香水、化妆品和珠宝购物网站:Brasty.bg
2020/04/22 全球购物
优秀高中生事迹材料
2014/02/11 职场文书
工会优秀工作者事迹
2014/08/17 职场文书
委托书的写法
2014/08/30 职场文书
2014年度工作总结报告
2014/12/15 职场文书
2015年小学开学寄语
2015/02/27 职场文书
MySQL GRANT用户授权的实现
2021/06/18 MySQL