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实现百度关键词排名查询
Mar 30 Python
Python实现的数据结构与算法之快速排序详解
Apr 22 Python
python处理html转义字符的方法详解
Jul 01 Python
python3.6 +tkinter GUI编程 实现界面化的文本处理工具(推荐)
Dec 20 Python
Python实现自动上京东抢手机
Feb 06 Python
Python实现繁体中文与简体中文相互转换的方法示例
Dec 18 Python
带你认识Django
Jan 15 Python
python实现爬虫抓取小说功能示例【抓取金庸小说】
Aug 09 Python
有关Tensorflow梯度下降常用的优化方法分享
Feb 04 Python
python GUI库图形界面开发之PyQt5布局控件QHBoxLayout详细使用方法与实例
Mar 06 Python
Django调用支付宝接口代码实例详解
Apr 04 Python
利用python制作拼图小游戏的全过程
Dec 04 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上的memcache和memcached两个pecl库
2010/03/29 PHP
Yii2.0表关联查询实例分析
2016/07/18 PHP
PHP实现PDO操作mysql存储过程示例
2019/02/13 PHP
php中目录操作opendir()、readdir()及scandir()用法示例
2019/06/08 PHP
实现动画效果核心方式的js代码
2013/09/27 Javascript
JavaScript显示当然日期和时间即年月日星期和时间
2013/10/29 Javascript
浅析javascript 定时器
2014/12/23 Javascript
JavaScript通过setTimeout实时显示当前时间的方法
2015/04/16 Javascript
AngularJS页面访问时出现页面闪烁问题的解决
2016/03/06 Javascript
利用JQuery直接调用asp.net后台的简单方法
2016/10/27 Javascript
微信小程序实现循环动画效果
2018/07/16 Javascript
JQuery常用简单动画操作方法回顾与总结
2019/12/07 jQuery
JS实现可控制的进度条
2020/03/25 Javascript
JavaScript 常见的继承方式汇总
2020/09/17 Javascript
详解python调度框架APScheduler使用
2017/03/28 Python
Django 忘记管理员或忘记管理员密码 重设登录密码的方法
2018/05/30 Python
Python实现的NN神经网络算法完整示例
2018/06/19 Python
python 字符串和整数的转换方法
2018/06/25 Python
Python+OpenCV实现图像融合的原理及代码
2018/12/03 Python
python flask搭建web应用教程
2019/11/19 Python
基于numpy中的expand_dims函数用法
2019/12/18 Python
浅谈在django中使用filter()(即对QuerySet操作)时踩的坑
2020/03/31 Python
python爬虫实例之获取动漫截图
2020/05/31 Python
python海龟绘图之画国旗实例代码
2020/11/11 Python
HTML5是否真的可以取代Flash
2010/02/10 HTML / CSS
美国在线家装零售商:Build.com
2016/09/02 全球购物
Linux常见面试题
2016/10/04 面试题
网站开发实习生的自我评价
2013/12/11 职场文书
打架检讨书400字
2014/01/17 职场文书
幼儿园教师教育感言
2014/02/28 职场文书
暑期培训班招生方案
2014/08/26 职场文书
学生检讨书如何写
2014/10/30 职场文书
2014年宣传部工作总结
2014/11/12 职场文书
2014年煤矿安全工作总结
2014/12/04 职场文书
学校教学工作总结2015
2015/05/19 职场文书
2016党员干部反腐倡廉心得体会
2016/01/13 职场文书