pytorch 如何自定义卷积核权值参数
更新时间：2021年05月27日 08:40:43 作者：Mr.Jcak
这篇文章主要介绍了pytorch 自定义卷积核权值参数的操作，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
pytorch中构建卷积层一般使用nn.Conv2d方法，有些情况下我们需要自定义卷积核的权值weight，而nn.Conv2d中的卷积参数是不允许自定义的，此时可以使用torch.nn.functional.conv2d简称F.conv2d
torch.nn.functional.conv2d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)
F.conv2d可以自己输入且也必须要求自己输入卷积权值weight和偏置bias。因此，构建自己想要的卷积核参数，再输入F.conv2d即可。
下面是一个用F.conv2d构建卷积层的例子
这里为了网络模型需要写成了一个类：
class CNN(nn.Module):
def __init__(self):
super(CNN, self).__init__()
self.weight = nn.Parameter(torch.randn(16, 1, 5, 5)) # 自定义的权值
self.bias = nn.Parameter(torch.randn(16)) # 自定义的偏置
def forward(self, x):
x = x.view(x.size(0), -1)
out = F.conv2d(x, self.weight, self.bias, stride=1, padding=0)
return out
值得注意的是，pytorch中各层需要训练的权重的数据类型设为nn.Parameter，而不是Tensor或者Variable。parameter的require_grad默认设置为true，而Varaible默认设置为False。
补充：pytorch中卷积参数的理解
kernel_size代表着卷积核，例如kernel_size=3或kernel_size=（3，7）；
stride：表明卷积核在像素级图像上行走的步长，如图2，步长为1；
padding：为上下左右填充的大小，例如padding=0/1/（1，1）/（1，3），
padding=0 不填充；
padding=1/（1，1） 上下左右分别填充1个格；
padding=（1，3） 高（上下）填充2个格，宽（左右）填充6个格；
卷积代码
torch.nn.Conv2d(512,512,kernel_size=(3,7),stride=2,padding=1)
指定输出形状的上采样
def upsample_add(self,x,y):
_,_,H,W = y.size()
return F.interpolate(x, size=(H,W), mode='bilinear', align_corners=False) + y
反卷积上采样
output_shape_w=kernel_size_w+(output_w-1)(kernel_size_w-1)+2padding
self.upscore2 = nn.ConvTranspose2d(
512, 1, kernel_size=3, stride=2,padding=0, bias=False)
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
pandas实现excel表格处理并读取指定sheet的方法
这篇文章主要介绍了pandas实现excel表格处理并读取指定sheet的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2024-02-02
这篇文章主要介绍了Windows环境中Python应用服务自启动及其监控问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-06-06
这篇文章主要为大家详细介绍了最全的Pycharm学习教程第四篇，Python解释器配置，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2017-05-05
今天在写代码的时候，需要统一化输出格式进行，一时想不起竟具体细节，用了最笨的方法，所以觉得有必要将常见的方法进行一个总结。下面这篇文中就给大家总结了python中常见的格式化输出，比如打印字符串、打印整数和打印浮点数等，下面来看看详细的输出方法吧。2016-12-12
python2与python3中关于对NaN类型数据的判断和转换方法
今天小编就为大家分享一篇python2与python3中关于对NaN类型数据的判断和转换方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-10-10
python中的逻辑运算符有两种返回值,python运算符除了能操作bool类型表达式,还能操作其他所有类型的表达式,这篇文章主要给大家介绍了关于python中and和or逻辑运算符用法的相关资料,需要的朋友可以参考下2022-01-01
本篇文章主要介绍了Python使用getpass库读取密码的示例，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-10-10
Python中的进程操作模块(multiprocess.process)
这篇文章介绍了Python中的进程操作模块(multiprocess.process)，文中通过示例代码介绍的非常详细。对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-05-05
这篇文章主要介绍了python中的正则表达式,贪婪匹配与非贪婪匹配方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-01-01
Python爬虫+tkinter界面实现历史天气查询的思路详解
这篇文章主要介绍了Python爬虫+tkinter界面实现历史天气查询的思路详解,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-02-02
最新评论