pytorch 实现在预训练模型的 input上增减通道
作者：Hi_AI 发布时间：2023-12-02 00:49:33
标签：pytorch,预训练模型,增减,通道
如何把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, :, :])
来源：https://blog.csdn.net/github_36923418/article/details/84567227
0
投稿猜你喜欢
读文件打开文件（文件需要存在）#打开文件f = open("data.txt","r")- Mysql安装、配置、优化，供大家参考，具体内容如下Mysql下载首先登入官网下载mysql的安装包，官网地址https://dev.mys
- 程序设计是困难的，其核心是管理的复杂性。计算机程序是人类做出的最复杂的东西。质量是不可靠的且隐蔽的。好的体系架构是必需给程序足够的结构使其健
- 然而，微软sql server在处理这类索引时，有个重要的缺陷，那就是把本该编译成索引seek的操作编成了索引扫描，这可能导致严重性能下降
- 这个问题的解决方案网上挺多的。其中我推荐的就是：with open(r"F:\Desktop\Book3.csv",&#
- 说到转置操作，顺便提及矩阵与数组的区别：矩阵：数学里的概念，其元素只能是数值，这也是区别于数组的根本所在数组：计算机中的概念，代表一种数据组
- 有一个查询如下： 代码如下:SELECT c.CustomerId, CompanyName FROM Customers c
- 首先，简单介绍一下EXECL中工作簿和工作表的区别：工作簿的英文是BOOK（WORKBOOK），工作表的英文是SHEET（WORKSHEET
- 一、什么是上下文管理器我们在处理文件的时候经常看到下面这样的代码，它即是上下文管理器：with open('test.txt'
- 一、自定义分页1、基础版自定义分页data = []for i in range(1, 302): tmp = {"i
- 目录urllib子模块request模块function request.urlopen()class request.Requestpyt
- 黑夜可能漫长，但总会迎来温暖的阳光，三月如期而至，武大的樱花又一次盛开。那么今天就一起来看看怎样在python中画一棵美丽的樱花树～说到用p
- 协程中未处理的异常会向上冒泡，传给 next 函数或 send 方法的调用方（即触发协程的对 象）。下面示例举例说明如何使用之前博客示例中由
- 看到论坛上有人模仿alert，自己也写了一个。本来想模仿winapi里的MessageBox ；但可惜js 不支持，阻塞模式。返回值只能用异
- 1. fixture的声明我们使用@pytest.fixture()来声明fixture函数。fixture()即可无参数进行声明，也可以带
- TXT文本文件，WORD文档点击后弹出另存为对话框，然后保存下载，而不是在浏览器中打开的asp实现方法，使用了asp中的stream对象，同
- V5.0之后，我们总结了一些得失。首先要说的是改版的动力。产品设计或产品升级的驱动力只有两个：用户需求和网站目标。之前的我们的多次改版，其驱
- 本文实例讲述了MySQL中USING 和 HAVING 用法。分享给大家供大家参考，具体如下：USING用于表连接时给定连接条件（可以理解为
- import导入模块import time #导入的时模块中的所有内容print(time.ctime()) #调用模块中的函数
- 1.软件环境Windows10 教育版64位Python 3.6.32.问题描述我们在定义一个函数或者是调用一个函数的时候，总是希望能够知道
