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 相关文章推荐
PyMongo安装使用笔记
Apr 27 Python
Python实现可获取网易页面所有文本信息的网易网络爬虫功能示例
Jan 15 Python
详解Django解决ajax跨域访问问题
Aug 24 Python
Python中分支语句与循环语句实例详解
Sep 13 Python
python批量复制图片到另一个文件夹
Sep 17 Python
Django数据库连接丢失问题的解决方法
Dec 29 Python
pytorch数据预处理错误的解决
Feb 20 Python
python3 配置logging日志类的操作
Apr 08 Python
Python+Kepler.gl实现时间轮播地图过程解析
Jul 20 Python
Python类成员继承重写的实现
Sep 16 Python
python利用tkinter实现图片格式转换的示例
Sep 28 Python
pytorch查看网络参数显存占用量等操作
May 12 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
全国FM电台频率大全 - 7 吉林省
2020/03/11 无线电
PHP读取zip文件的方法示例
2016/11/17 PHP
jquery animate图片模向滑动示例代码
2011/01/26 Javascript
DWZ table的原生分页浅谈
2013/03/01 Javascript
javascript重写alert方法的实例代码
2013/03/29 Javascript
JS数组(Array)处理函数整理
2014/12/07 Javascript
javascript+HTML5的Canvas实现Lab单车动画效果
2015/08/07 Javascript
AngularJS基础 ng-keydown 指令简单示例
2016/08/02 Javascript
jQuery选择器总结之常用元素查找方法
2016/08/04 Javascript
vue 请求后台数据的实例代码
2017/06/22 Javascript
vue router-link传参以及参数的使用实例
2017/11/10 Javascript
浅谈vue的props,data,computed变化对组件更新的影响
2018/01/16 Javascript
通过webpack引入第三方库的方法
2018/07/20 Javascript
NodeJs实现简单的爬虫功能案例分析
2018/12/05 NodeJs
vue.js实现h5机器人聊天(测试版)
2020/07/16 Javascript
详解微信小程序轨迹回放实现及遇到的坑
2021/02/02 Javascript
[59:07]海涛为你详解DOTA2新版本“贤哲秘契”
2014/11/22 DOTA
零基础写python爬虫之使用urllib2组件抓取网页内容
2014/11/04 Python
Python读大数据txt
2016/03/28 Python
Pandas 同元素多列去重的实例
2018/07/03 Python
python实现彩票系统
2020/06/28 Python
Python使用pickle模块报错EOFError Ran out of input的解决方法
2018/08/16 Python
解决PySide+Python子线程更新UI线程的问题
2019/01/11 Python
对Python中画图时候的线类型详解
2019/07/07 Python
带你彻底搞懂python操作mysql数据库(cursor游标讲解)
2020/01/06 Python
python模式 工厂模式原理及实例详解
2020/02/11 Python
pycharm解决关闭flask后依旧可以访问服务的问题
2020/04/03 Python
Python 实现一行输入多个数字(用空格隔开)
2020/04/29 Python
HTML5实现页面切换激活的PageVisibility API使用初探
2016/05/13 HTML / CSS
皇家阿尔伯特英国官方商店:Royal Albert骨瓷
2019/03/25 全球购物
SOKOLOV官网:俄罗斯珠宝首饰品牌
2021/01/02 全球购物
《在家里》教后反思
2014/03/01 职场文书
网上祭英烈活动总结
2015/02/04 职场文书
2015年暑假工作总结
2015/07/13 职场文书
python用字节处理文件实例讲解
2021/04/13 Python
详细介绍MySQL中limit和offset的用法
2022/05/06 MySQL