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的lambda匿名函数的简单介绍
Apr 25 Python
Python的组合模式与责任链模式编程示例
Feb 02 Python
理解Python中的With语句
Mar 18 Python
Python用imghdr模块识别图片格式实例解析
Jan 11 Python
python实现画圆功能
Jan 25 Python
Python3内置模块random随机方法小结
Jul 13 Python
python实现代码统计程序
Sep 19 Python
Python使用Socket实现简单聊天程序
Feb 28 Python
五种Python转义表示法
Nov 27 Python
python实现登录与注册系统
Nov 30 Python
python中delattr删除对象方法的代码分析
Dec 15 Python
Python移位密码、仿射变换解密实例代码
Jun 27 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
php实现的Timer页面运行时间监测类
2014/09/24 PHP
php的curl封装类用法实例
2014/11/07 PHP
js 实现复制到粘贴板的功能代码
2010/05/13 Javascript
验证javascript中Object和Function的关系的三段简单代码
2010/06/27 Javascript
extJS中常用的4种Ajax异步提交方式
2014/03/07 Javascript
javascript在当前窗口关闭前检测窗口是否关闭
2014/09/29 Javascript
JavaScript将字符串转换为整数的方法
2015/04/14 Javascript
基于Jquery+div+css实现弹出登录窗口(代码超简单)
2015/10/27 Javascript
基于Taro的微信小程序模板消息-获取formId功能模块封装实践
2019/07/15 Javascript
vue+vant实现商品列表批量倒计时功能
2020/01/13 Javascript
JS实现transform实现扇子效果
2020/01/17 Javascript
[53:13]2014 DOTA2国际邀请赛中国区预选赛5.21 DT VS LGD-GAMING
2014/05/22 DOTA
对python中两种列表元素去重函数性能的比较方法
2018/06/29 Python
浅谈Python中的bs4基础
2018/10/21 Python
Django 中间键和上下文处理器的使用
2019/03/17 Python
python使用pymongo操作mongo的完整步骤
2019/04/13 Python
Python处理时间日期坐标轴过程详解
2019/06/25 Python
python 实现生成均匀分布的点
2019/12/05 Python
Python3 把一个列表按指定数目分成多个列表的方式
2019/12/25 Python
Python打开文件、文件读写操作、with方式、文件常用函数实例分析
2020/01/07 Python
通过自学python能找到工作吗
2020/06/21 Python
Python如何优雅删除字符列表空字符及None元素
2020/06/25 Python
html5记忆翻牌游戏实现思路及代码
2013/07/25 HTML / CSS
html5实现多图片预览上传及点击可拖拽控件
2018/03/15 HTML / CSS
UGG澳洲官网:UGG Australia
2018/04/26 全球购物
美国领先的宠物用品和宠物食品零售商:Petco
2020/10/28 全球购物
一些PHP的面试题
2015/05/06 面试题
大学生就业自荐信
2013/10/26 职场文书
ktv收银员岗位职责
2013/12/16 职场文书
管理部副部长岗位职责范文
2014/03/09 职场文书
新年爱情寄语
2014/04/08 职场文书
大学竞选班长演讲稿
2014/04/24 职场文书
拒绝黄毒毒宣传标语
2014/06/26 职场文书
人大调研汇报材料
2014/08/14 职场文书
2014领导班子正风肃纪思想汇报
2014/09/18 职场文书
生日祝酒词大全
2015/08/10 职场文书