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的re模块应用实例
Sep 26 Python
在Python下尝试多线程编程
Apr 28 Python
10分钟教你用Python实现微信自动回复功能
Nov 28 Python
Python3 jupyter notebook 服务器搭建过程
Nov 30 Python
Python模块的加载讲解
Jan 15 Python
python序列化与数据持久化实例详解
Dec 20 Python
python实现梯度法 python最速下降法
Mar 24 Python
Keras之fit_generator与train_on_batch用法
Jun 17 Python
Python读取xlsx数据生成图标代码实例
Aug 12 Python
PyCharm 2020.2.2 x64 下载并安装的详细教程
Oct 15 Python
python中使用np.delete()的实例方法
Feb 01 Python
Pytorch中TensorBoard及torchsummary的使用详解
May 12 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
腾讯QQ微博API接口获取微博内容
2013/10/30 PHP
codeigniter使用技巧批量插入数据实例方法分享
2013/12/31 PHP
详解PHP中的状态模式编程
2015/08/11 PHP
php基于环形链表解决约瑟夫环问题示例
2017/11/07 PHP
比较简单实用的使用正则三种版本的js去空格处理方法
2007/11/18 Javascript
使用JavaScript构建JSON格式字符串实现步骤
2013/03/22 Javascript
A标签触发onclick事件而不跳转的多种解决方法
2013/06/27 Javascript
实现只能输入数字的input不用replace方法
2013/09/12 Javascript
JS OffsetParent属性深入解析
2014/01/13 Javascript
jQuery中parents()方法用法实例
2015/01/07 Javascript
javascript cookie的简单应用
2016/02/24 Javascript
Bootstrap项目实战之首页内容介绍(全)
2016/04/25 Javascript
JavaScript的==运算详解
2016/07/20 Javascript
javascript对浅拷贝和深拷贝的详解
2016/10/14 Javascript
jQuery Easyui datagrid连续发送两次请求问题
2016/12/13 Javascript
canvas滤镜效果实现代码
2017/02/06 Javascript
微信小程序实现之手势锁功能实例代码
2018/07/19 Javascript
JS实现根据详细地址获取经纬度功能示例
2019/04/16 Javascript
JavaScript对象原型链原理详解
2020/02/05 Javascript
react国际化化插件react-i18n-auto使用详解
2020/03/31 Javascript
VUE中V-IF条件判断改变元素的样式操作
2020/08/09 Javascript
使用Mock.js生成前端测试数据
2020/12/13 Javascript
[04:13]2014DOTA2国际邀请赛 专访DC目前形势不容乐观
2014/07/12 DOTA
python中查找excel某一列的重复数据 剔除之后打印
2013/02/10 Python
在django模板中实现超链接配置
2019/08/21 Python
Python CSS选择器爬取京东网商品信息过程解析
2020/06/01 Python
CSS3中的@keyframes关键帧动画的选择器绑定
2016/06/13 HTML / CSS
HTML5 history新特性pushState、replaceState及两者的区别
2015/12/26 HTML / CSS
H5 meta小结(前端必看篇)
2016/08/24 HTML / CSS
HTML5拖拽功能实现的拼图游戏
2018/07/31 HTML / CSS
HTML5 3D书本翻页动画的实现示例
2019/08/28 HTML / CSS
Crocs卡骆驰洞洞鞋日本官方网站:Crocs日本
2016/08/25 全球购物
澳大利亚优惠网站:Deals.com.au
2019/07/02 全球购物
输入一行文字,找出其中大写字母、小写字母、空格、数字、及其他字符各有多少
2016/04/15 面试题
原料仓管员岗位职责
2015/04/01 职场文书
一文了解MySQL二级索引的查询过程
2022/02/24 MySQL