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生成随机mac地址的方法
Mar 16 Python
分享一下Python 开发者节省时间的10个方法
Oct 02 Python
python解决Fedora解压zip时中文乱码的方法
Sep 18 Python
详解python中executemany和序列的使用方法
Aug 12 Python
Python使用回溯法子集树模板解决爬楼梯问题示例
Sep 08 Python
Python SqlAlchemy动态添加数据表字段实例解析
Feb 07 Python
pandas分区间,算频率的实例
Jul 04 Python
python3安装crypto出错及解决方法
Jul 30 Python
Python使用scrapy爬取阳光热线问政平台过程解析
Aug 14 Python
Python实现word2Vec model过程解析
Dec 16 Python
django ajax发送post请求的两种方法
Jan 05 Python
python 连续不等式语法糖实例
Apr 15 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
如何使用jQuery+PHP+MySQL来实现一个在线测试项目
2015/04/26 PHP
PHP YII框架开发小技巧之模型(models)中rules自定义验证规则
2015/11/16 PHP
Discuz不使用插件实现简单的打赏功能
2019/03/21 PHP
JavaScript学习历程和心得小结
2010/08/16 Javascript
ajax 同步请求和异步请求的差异分析
2011/07/04 Javascript
javascript 基础篇4 window对象,DOM
2012/03/14 Javascript
javascript实现博客园页面右下角返回顶部按钮
2015/02/22 Javascript
JavaScript实现将数组中所有元素连接成一个字符串的方法
2015/04/06 Javascript
javascript实现图片自动和可控的轮播切换特效
2015/04/13 Javascript
JavaScript实现的背景自动变色代码
2015/10/17 Javascript
jQuery中text() val()和html()的区别实例详解
2016/06/28 Javascript
原生js获取iframe中dom元素--父子页面相互获取对方dom元素的方法
2016/08/05 Javascript
vue中将网页打印成pdf实例代码
2017/06/15 Javascript
浅析Vue自定义组件的v-model
2017/11/26 Javascript
JavaScript寄生组合式继承实例详解
2018/01/06 Javascript
angular 服务的单例模式(依赖注入模式下)详解
2018/10/22 Javascript
vite2.0+vue3移动端项目实战详解
2021/03/03 Vue.js
Python实现动态添加类的属性或成员函数的解决方法
2014/07/16 Python
Python标准库之sqlite3使用实例
2014/11/25 Python
Python实现的爬虫功能代码
2017/06/24 Python
python3.6+opencv3.4实现鼠标交互查看图片像素
2018/02/26 Python
Django项目开发中cookies和session的常用操作分析
2018/07/03 Python
python将list转为matrix的方法
2018/12/12 Python
Python面向对象实现一个对象调用另一个对象操作示例
2019/04/08 Python
通过python调用adb命令对App进行性能测试方式
2020/04/23 Python
Windows 下更改 jupyterlab 默认启动位置的教程详解
2020/05/18 Python
python 如何在测试中使用 Mock
2021/03/01 Python
我想声明一个指针并为它分配一些空间, 但却不行。这些代码有什么 问题?char *p; *p = malloc(10);
2016/10/06 面试题
大学毕业生简单自荐信
2013/11/05 职场文书
市场营销专科应届生求职信
2013/11/24 职场文书
商务代表岗位职责
2015/02/15 职场文书
2015年爱国卫生月活动总结
2015/03/26 职场文书
证劵公司反洗钱宣传活动总结
2015/05/08 职场文书
阿凡达观后感
2015/06/10 职场文书
通讯稿范文
2015/07/22 职场文书
Centos7 Shell编程之正则表达式、文本处理工具详解
2022/08/05 Servers