pytorch 实现在预训练模型的 input上增减通道
更新时间：2020年01月06日 11:31:51 作者：Hi_AI
今天小编就为大家分享一篇pytorch 实现在预训练模型的 input上增减通道，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
如何把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上增减通道就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
OpenPyXL是一个强大的Python库,用于处理Excel文件,允许读取、编辑和创建Excel工作簿和工作表,本文将详细介绍OpenPyXL的各种功能,希望对大家有所帮助2023-11-11
这篇文章主要介绍了python,Django实现的淘宝客登录功能,结合实例形式分析了Django框架基于淘宝接口的登录功能相关操作技巧,需要的朋友可以参考下2019-06-06
这篇文章主要介绍了Python垃圾回收机制三种实现方法,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-04-04
在 Python 中,多进程编程是一种提高程序运行效率的有效手段,本文将详细讲解 Python 多进程编程的基本概念,使用方法及注意事项,希望对大家有所帮助2024-12-12
这篇文章主要介绍了如何基于Python语言实现遍历多个子文件夹，将每一个子文件夹中大量的文件，按照每一个文件的文件名称的特点复制到不同的目标文件夹中，感兴趣的可以了解下2023-08-08
这篇文章主要介绍了Python如何测试stdout输出，帮助大家更好的理解和学习Python，感兴趣的朋友可以了解下2020-08-08
这篇文章主要介绍了Python中asyncore的用法,asyncore提供了方便的网络操作方法,本文以连接并解析www.python.org主页为例加以说明,需要的朋友可以参考下2014-09-09
这篇文章主要介绍了python动态加载技术解析,说简单点就是，如果开发者发现自己的代码有bug，那么他可以在不关闭原来代码的基础之上，动态替换模块替换方法一般用reload来完成,需要的朋友可以参考下2023-07-07
今天小编就为大家分享一篇Python代码打开本地.mp4格式文件的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
本文给大家汇总介绍了Python安装第三方库及常见问题处理方法，非常的简单使用，有需要的小伙伴可以参考下2016-09-09
最新评论