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)

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python实现手势识别
这篇文章主要介绍了Python如何实现手指指尖的检测，并且可以在windows系统下通过判断手指数目，来模拟键盘操作，感兴趣的朋友可以了解下
2020-10-10
Python中matplotlib如何改变画图的字体
这篇文章主要介绍了Python中matplotlib如何改变画图的字体,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-08-08
Python实现控制手机电脑拍照并自动发送邮箱
这篇文章主要介绍了如何实现利用Python控制手机电脑拍照并自动发送邮箱，文中的示例代码讲解详细，感兴趣的小伙伴可以跟随小编一起动手试一试
2022-02-02
python 使用xlsxwriter循环向excel中插入数据和图片的操作
这篇文章主要介绍了python 使用xlsxwriter循环向excel中插入数据和图片的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2021-01-01
Python中Qslider控件实操详解
在本篇文章里小编给大家整理的是一篇关于Python中Qslider控件实操详解内容，对此有兴趣的朋友们可以跟着学习参考下。
2021-02-02
pytorch 查看cuda 版本方式
这篇文章主要介绍了pytorch 查看cuda 版本方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-06-06
Python:一行代码,导入Python所有库
在本篇内容里小编给大家分享如何用一行代码,导入Python所有库，有需要的朋友们可以学习下，希望能够给你带来帮助
2021-10-10
Python正则表达式如何进行字符串替换实例
Python正则表达式在使用中会经常应用到字符串替换的代码。这篇文章主要介绍了Python正则表达式如何进行字符串替换,具有一定的参考价值，感兴趣的小伙伴们可以参考一下。
2016-12-12
python 包中的sched 事件调度器的操作方法
sched模块内容很简单，只定义了一个类。它用来最为一个通用的事件调度模块，接下来通过本文给大家介绍python 包之 sched 事件调度器教程,需要的朋友可以参考下
2022-04-04
Python标准库pickle的简单使用
本文主要介绍了Python标准库pickle的简单使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2023-05-05