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上增减通道就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了python使用装饰器和线程限制函数执行时间的方法,主要涉及timelimited函数的使用技巧,非常具有实用价值,需要的朋友可以参考下2015-04-04
tempfile模块专门用于创建临时文件和临时目录，它既可以在 UNIX 平台上运行良好，也可以在 Windows 平台上运行良好。本文将利用tempfile模块创建临时文件来保存数据，感兴趣的可以了解一下2022-07-07
这篇文章主要介绍了Python subprocess库的使用详解，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-10-10
这篇文章主要介绍了如何利用Python对Word内容进行各种样式的设置，让其能够看起来更加的美观。文中的示例代码讲解详细，需要的可以参考一下2022-05-05
在本文总小编给大家整理了关于python之中的迭代与迭代对象的相关基础知识点内容，需要的朋友们学习下。2018-10-10
这篇文章主要为大家详细介绍了python远程连接MySQL数据库，拉取数据存至本地文件，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-04-04
这篇文章主要介绍了Python读写Excel文件方法介绍,本文讲解了xlrd、xlwt、xlutils等类库的使用,需要的朋友可以参考下2014-11-11
在本篇文章里小编给大家整理的是关于如何用python免费看美剧的方法内容，需要的朋友们可以学习下。2020-08-08
这篇文章主要为大家详细介绍了基于python3的socket聊天编程，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-02-02
这篇文章主要介绍了Python中基本类型的连接组合和互相转换13种方式,本文给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下2019-12-12
最新评论