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之for循环语句
Oct 02 Python
python复制与引用用法分析
Apr 08 Python
Python CSV模块使用实例
Apr 09 Python
Python 专题六 局部变量、全局变量global、导入模块变量
Mar 20 Python
python 巧用正则寻找字符串中的特定字符的位置方法
May 02 Python
浅谈Tensorflow由于版本问题出现的几种错误及解决方法
Jun 13 Python
python使用Plotly绘图工具绘制散点图、线形图
Apr 02 Python
基于python爬取有道翻译过程图解
Mar 31 Python
Python异常处理机制结构实例解析
Jul 23 Python
Python中实现一行拆多行和多行并一行的示例代码
Sep 06 Python
Python并发爬虫常用实现方法解析
Nov 19 Python
Python机器学习之基于Pytorch实现猫狗分类
Jun 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
JavaScript 操作键盘的Enter事件(键盘任何事件),兼容多浏览器
2010/10/11 Javascript
ExtJS 入门
2010/10/29 Javascript
js中indexof的用法详细解析
2013/12/24 Javascript
浅析jquery某一元素重复绑定的问题
2014/01/03 Javascript
jQuery中has()方法用法实例
2015/01/06 Javascript
JS版元素周期表实现方法
2015/08/05 Javascript
全面解析bootstrap格子布局
2016/05/22 Javascript
JavaScript与JQUERY获取元素的宽、高和位置
2017/02/26 Javascript
Angular+Node生成随机数的方法
2017/06/16 Javascript
javascript编程开发中取色器及封装$函数用法示例
2017/08/09 Javascript
新手vue构建单页面应用实例代码
2017/09/18 Javascript
JS实现提交表单前的数字及邮箱校检功能
2017/11/13 Javascript
Mac中安装nvm的教程分享
2017/12/11 Javascript
Webpack4+Babel7+ES6兼容IE8的实现
2019/04/10 Javascript
vue-router 前端路由之路由传值的方式详解
2019/04/30 Javascript
javascript实现的时间格式加8小时功能示例
2019/06/13 Javascript
如何通过vscode运行调试javascript代码
2020/07/24 Javascript
详解datagrid使用方法(重要)
2020/11/06 Javascript
[51:53]完美世界DOTA2联赛决赛日 Inki vs LBZS 第二场 11.08
2020/11/10 DOTA
编写Python脚本来获取Google搜索结果的示例
2015/05/04 Python
举例简单讲解Python中的数据存储模块shelve的用法
2016/03/03 Python
Python实现定时自动关闭的tkinter窗口方法
2019/02/16 Python
python算法题 链表反转详解
2019/07/02 Python
Python多线程threading join和守护线程setDeamon原理详解
2020/03/18 Python
Python web框架(django,flask)实现mysql数据库读写分离的示例
2020/11/18 Python
实例讲解利用HTML5 Canvas API操作图形旋转的方法
2016/03/22 HTML / CSS
参观监狱心得体会
2014/01/02 职场文书
《哪吒闹海》教学反思
2014/02/28 职场文书
监督检查工作方案
2014/05/28 职场文书
服装设计师求职信
2014/06/04 职场文书
绿色出行口号
2014/06/18 职场文书
爱护草坪标语
2014/06/24 职场文书
2015上半年个人工作总结
2015/07/27 职场文书
思品教学工作总结
2015/08/10 职场文书
《烈火英雄》观后感:致敬和平时代的英雄
2019/11/11 职场文书
一次MySQL启动导致的事故实战记录
2021/09/15 MySQL