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 相关文章推荐
Python3里的super()和__class__使用介绍
Apr 23 Python
Python中动态创建类实例的方法
Mar 24 Python
Python中enumerate()函数编写更Pythonic的循环
Mar 06 Python
tensorflow 用矩阵运算替换for循环 用tf.tile而不写for的方法
Jul 27 Python
Python设计模式之模板方法模式实例详解
Jan 17 Python
解决django中ModelForm多表单组合的问题
Jul 18 Python
python 3.74 运行import numpy as np 报错lib\site-packages\numpy\__init__.py
Oct 06 Python
用python介绍4种常用的单链表翻转的方法小结
Feb 24 Python
python如何利用paramiko执行服务器命令
Nov 07 Python
python 通过 pybind11 使用Eigen加速代码的步骤
Dec 07 Python
用Python实现定时备份Mongodb数据并上传到FTP服务器
Jan 27 Python
Pyside2中嵌入Matplotlib的绘图的实现
Feb 22 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
咖啡冲泡指南 咖啡有哪些制作方式 单品咖啡 意式咖啡
2021/03/06 冲泡冲煮
php 代码优化的42条建议 推荐
2009/09/25 PHP
PHP+MySQL 制作简单的留言本
2009/11/02 PHP
php fputcsv命令 写csv文件遇到的小问题(多维数组连接符)
2011/05/24 PHP
采用PHP函数memory_get_usage获取PHP内存清耗量的方法
2011/12/06 PHP
PHP 自动加载的简单实现(推荐)
2016/08/12 PHP
使用SMB共享来绕过php远程文件包含的限制执行RFI的利用
2019/05/31 PHP
如何解决PHP获取不到SESSION信息之一般情况
2019/10/10 PHP
列表内容的选择
2006/06/30 Javascript
16个最流行的JavaScript框架[推荐]
2011/05/29 Javascript
jquery实现奇偶行赋值不同css值
2012/02/17 Javascript
JS验证控制输入中英文字节长度(input、textarea等)具体实例
2013/06/21 Javascript
jquery 获取标签名(tagName)示例代码
2013/07/11 Javascript
JavaScript 模块化编程(笔记)
2015/04/08 Javascript
javascript基础知识分享之类与函数化
2016/02/13 Javascript
深入理解jQuery layui分页控件的使用
2016/08/17 Javascript
Angular2自定义分页组件
2017/04/19 Javascript
ng-events类似ionic中Events的angular全局事件
2018/09/05 Javascript
js 将线性数据转为树形的示例代码
2019/05/28 Javascript
详解Nuxt.js 实战集锦
2019/11/19 Javascript
JS倒计时两种实现方式代码实例
2020/07/27 Javascript
uniapp实现可滑动选项卡
2020/10/21 Javascript
python数据分析数据标准化及离散化详解
2018/02/26 Python
浅谈Python中的私有变量
2018/02/28 Python
浅析python实现scrapy定时执行爬虫
2018/03/04 Python
Python中出现IndentationError:unindent does not match any outer indentation level错误的解决方法
2020/04/18 Python
使用Django简单编写一个XSS平台的方法步骤
2019/03/25 Python
Python 2/3下处理cjk编码的zip文件的方法
2019/04/26 Python
Python安装selenium包详细过程
2019/07/23 Python
python selenium 查找隐藏元素 自动播放视频功能
2019/07/24 Python
Python selenium自动化测试模型图解
2020/04/15 Python
HTML5通过调用canvas对象的getContext()方法来获取绘图环境
2014/06/23 HTML / CSS
三星法国官方网站:Samsung法国
2019/10/31 全球购物
关于.NET, HTML的五个问题
2012/08/29 面试题
鸡毛信观后感
2015/06/11 职场文书
使用HttpSessionListener监听器实战
2022/03/17 Java/Android