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标准日志模块logging的使用方法
Nov 01 Python
详解Python的Twisted框架中reactor事件管理器的用法
May 25 Python
python 类详解及简单实例
Mar 24 Python
python爬虫入门教程--正则表达式完全指南(五)
May 25 Python
Python正则捕获操作示例
Aug 19 Python
python贪婪匹配以及多行匹配的实例讲解
Apr 19 Python
python抽取指定url页面的title方法
May 11 Python
Python日志模块logging基本用法分析
Aug 23 Python
在python里使用await关键字来等另外一个协程的实例
May 04 Python
Python Pillow(PIL)库的用法详解
Sep 19 Python
Python制作简单的剪刀石头布游戏
Dec 10 Python
python3使用diagrams绘制架构图的步骤
Apr 08 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
一台收音机,让一家人都笑逐颜开!
2020/08/21 无线电
PHP默认安装产生系统漏洞
2006/10/09 PHP
PHP面向对象——访问修饰符介绍
2012/11/08 PHP
PHP排序二叉树基本功能实现方法示例
2018/05/26 PHP
PHP中localeconv()函数的用法
2019/03/26 PHP
javascript 最常用的10个自定义函数[推荐]
2009/12/26 Javascript
JS是按值传递还是按引用传递
2015/01/30 Javascript
JavaScript实现Java中StringBuffer的方法
2015/02/09 Javascript
JavaScript获取页面中表单(form)数量的方法
2015/04/03 Javascript
JS数字抽奖游戏实现方法
2015/05/04 Javascript
jQuery实现可高亮显示的二级CSS菜单效果
2015/09/01 Javascript
js实现微信分享代码
2020/10/11 Javascript
精通JavaScript的this关键字
2020/05/28 Javascript
浅析C/C++,Java,PHP,JavaScript,Json数组、对象赋值时最后一个元素后面是否可以带逗号
2016/03/22 Javascript
jQuery插件DataTables分页开发心得体会
2017/08/22 jQuery
从vue源码看props的用法
2019/01/09 Javascript
Vue使用vue-draggable 插件在不同列表之间拖拽功能
2020/03/12 Javascript
swiperjs实现导航与tab页的联动
2020/12/13 Javascript
Python 字符串操作实现代码(截取/替换/查找/分割)
2013/06/08 Python
urllib和BeautifulSoup爬取维基百科的词条简单实例
2018/01/17 Python
PyCharm代码整体缩进,反向缩进的方法
2018/06/25 Python
利用ctypes获取numpy数组的指针方法
2019/02/12 Python
python topk()函数求最大和最小值实例
2020/04/02 Python
css3实现3d旋转动画特效
2015/03/10 HTML / CSS
CSS3 @media的基本用法总结
2019/09/10 HTML / CSS
Web时代变迁及html5与html4的区别
2016/01/06 HTML / CSS
美国著名的团购网站:Woot
2016/08/02 全球购物
创业计划书的内容步骤和要领
2014/01/04 职场文书
代理商会议邀请函
2014/01/27 职场文书
化妆品促销方案
2014/02/24 职场文书
幼儿园亲子活动总结
2014/04/26 职场文书
2014年机关作风建设工作总结
2014/10/23 职场文书
校车司机安全责任书
2015/05/11 职场文书
《追风筝的人》:人心中的成见是座大山,但请不忘初心
2019/11/15 职场文书
python 如何获取页面所有a标签下href的值
2021/05/06 Python
Jpa Specification如何实现and和or同时使用查询
2021/11/23 Java/Android