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登陆asp网站页面的实现代码
Jan 14 Python
在Python上基于Markov链生成伪随机文本的教程
Apr 17 Python
python获取当前日期和时间的方法
Apr 30 Python
学习python之编写简单简单连接数据库并执行查询操作
Feb 27 Python
Python实现KNN邻近算法
Jan 28 Python
python获取命令行输入参数列表的实例代码
Jun 23 Python
Pytorch在NLP中的简单应用详解
Jan 08 Python
Python pandas对excel的操作实现示例
Jul 21 Python
Python 在局部变量域中执行代码
Aug 07 Python
Java Unsafe类实现原理及测试代码
Sep 15 Python
python利用platform模块获取系统信息
Oct 09 Python
python中的getter与setter你了解吗
Mar 24 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
thinkPHP模板算术运算相关函数用法分析
2016/07/12 PHP
php版阿里大于(阿里大鱼)短信发送实例详解
2016/11/30 PHP
使用vs code编辑调试php配置的方法
2019/01/29 PHP
浅析jQuery对select操作小结(遍历option,操作option)
2013/07/04 Javascript
JS的Document属性和方法小结
2013/09/17 Javascript
利用JS来控制键盘的上下左右键(示例代码)
2013/12/14 Javascript
Jquery.Form 异步提交表单的简单实例
2014/03/03 Javascript
浅析javascript操作 cookie对象
2014/12/26 Javascript
javascript元素动态创建实现方法
2015/05/13 Javascript
在JavaScript的AngularJS库中进行单元测试的方法
2015/06/23 Javascript
jQuery实现浮动层随浏览器滚动条滚动的方法
2015/09/22 Javascript
jQuery简单实现提交数据出现loading进度条的方法
2016/03/29 Javascript
Vue.js结合bootstrap实现分页控件
2017/03/10 Javascript
layui表格 返回的数据状态异常的解决方法
2019/09/10 Javascript
vue实现分页加载效果
2019/12/24 Javascript
[02:15]2015国际邀请赛选手档案IG.Ferrari 430
2015/07/30 DOTA
python执行get提交的方法
2015/04/29 Python
详解使用python crontab设置linux定时任务
2016/12/08 Python
python用pickle模块实现“增删改查”的简易功能
2017/06/07 Python
python调用百度REST API实现语音识别
2018/08/30 Python
python3实现绘制二维点图
2019/12/04 Python
python opencv 图像边框(填充)添加及图像混合的实现方法(末尾实现类似幻灯片渐变的效果)
2020/03/09 Python
python3用urllib抓取贴吧邮箱和QQ实例
2020/03/10 Python
Python 使用生成器代替线程的方法
2020/08/04 Python
荷兰领先的百货商店:De Bijenkorf
2018/10/17 全球购物
27个经典Linux面试题及答案,你知道几个?
2013/01/10 面试题
财务会计专业推荐信
2013/11/30 职场文书
魅力教师事迹材料
2014/01/10 职场文书
咖啡书吧创业计划书
2014/01/13 职场文书
高三生物教学反思
2014/01/25 职场文书
初中毕业生的自我评价
2014/03/03 职场文书
学生个人自我鉴定范文
2014/03/28 职场文书
英语教师求职信
2014/06/16 职场文书
2014年乡镇妇联工作总结
2014/12/02 职场文书
八年级作文之我的母亲
2019/12/10 职场文书
Nginx反向代理多个服务器的实现方法
2021/03/31 Servers