pytorch 实现在预训练模型的 input上增减通道


Posted in Python onJanuary 06, 2020

如何把imagenet预训练的模型,输入层的通道数随心所欲的修改,从而来适应自己的任务

#增加一个通道
w = layers[0].weight
layers[0] = nn.Conv2d(4, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)
layers[0].weight = torch.nn.Parameter(torch.cat((w, w[:, :1, :, :]), dim=1))
 
#方式2
w = layers[0].weight
layers[0] = nn.Conv2d(4, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)
layers[0].weight = torch.nn.Parameter(torch.cat((w, torch.zeros(64, 1, 7, 7)), dim=1))
 
 
#单通道输入
layers[0] = nn.Conv2d(1, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)
layers[0].weight = torch.nn.Parameter(w[:, :1, :, :])

以上这篇pytorch 实现在预训练模型的 input上增减通道就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python中redis的安装和使用
Dec 04 Python
python opencv之SURF算法示例
Feb 24 Python
详谈在flask中使用jsonify和json.dumps的区别
Mar 26 Python
python爬虫 使用真实浏览器打开网页的两种方法总结
Apr 21 Python
Python之list对应元素求和的方法
Jun 28 Python
python实现排序算法解析
Sep 08 Python
如何用python写一个简单的词法分析器
Dec 18 Python
详解Python 解压缩文件
Apr 09 Python
Python如何优雅删除字符列表空字符及None元素
Jun 25 Python
Python图像读写方法对比
Nov 16 Python
Python3自带工具2to3.py 转换 Python2.x 代码到Python3的操作
Mar 03 Python
在pyCharm中下载第三方库的方法
Apr 18 Python
Python 将json序列化后的字符串转换成字典(推荐)
Jan 06 #Python
Python tkinter三种布局实例详解
Jan 06 #Python
Python有参函数使用代码实例
Jan 06 #Python
python通过移动端访问查看电脑界面
Jan 06 #Python
基于pytorch 预训练的词向量用法详解
Jan 06 #Python
导入tensorflow:ImportError: libcublas.so.9.0 报错
Jan 06 #Python
Python连接字符串过程详解
Jan 06 #Python
You might like
sqlyog 中文乱码问题的设置方法
2008/10/19 PHP
PHP 转义使用详解
2013/07/15 PHP
ThinkPHP3.1新特性之对页面压缩输出的支持
2014/06/19 PHP
php使用CURL模拟GET与POST向微信接口提交及获取数据的方法
2016/09/23 PHP
在laravel中使用Symfony的Crawler组件分析HTML
2017/06/19 PHP
php封装的pdo数据库操作工具类与用法示例
2019/05/08 PHP
Laravel框架表单验证操作实例分析
2019/09/30 PHP
JavaScript列表框listbox全选和反选的实现方法
2015/03/18 Javascript
文字垂直滚动之javascript代码
2015/07/29 Javascript
JavaScript中的this关键字使用详解
2015/08/14 Javascript
jQuery中的ready函数与window.onload谁先执行
2016/06/21 Javascript
JS 动态判断PC和手机浏览器实现代码
2016/09/21 Javascript
jquery实现百叶窗效果
2017/01/12 Javascript
js实现文本上下来回滚动
2017/02/03 Javascript
Angular.js中上传指令ng-upload的基本使用教程
2017/07/30 Javascript
详解用Node.js写一个简单的命令行工具
2018/03/01 Javascript
如何在Vue中使用CleaveJS格式化你的输入内容
2018/12/14 Javascript
微信小程序动画组件使用解析,类似vue,且更强大
2019/08/01 Javascript
使用 webpack 插件自动生成 vue 路由文件的方法
2019/08/20 Javascript
浅谈对于“不用setInterval,用setTimeout”的理解
2019/08/28 Javascript
python用ConfigObj读写配置文件的实现代码
2013/03/04 Python
python操作redis的方法
2015/07/07 Python
Python脚本处理空格的方法
2016/08/08 Python
Python爬取网易云音乐热门评论
2017/03/31 Python
tensorboard实现同时显示训练曲线和测试曲线
2020/01/21 Python
详解python with 上下文管理器
2020/09/02 Python
canvas实现圆绘制的示例代码
2019/09/11 HTML / CSS
Merrell美国官网:美国登山运动鞋品牌
2018/02/07 全球购物
墨西哥皇宫度假村预订:Palace Resorts
2018/06/16 全球购物
店面销售职位的职责
2014/03/09 职场文书
《风娃娃》教学反思
2014/04/19 职场文书
反腐倡廉警示教育活动心得体会
2014/09/04 职场文书
2015年度销售个人工作总结
2015/03/31 职场文书
2015年学习部工作总结范文
2015/03/31 职场文书
《月光曲》教学反思
2016/02/16 职场文书
超市啤酒狂欢夜策划方案范文!
2019/07/03 职场文书