图解卷积层stride,padding,kernel_size 和卷积前后特征图尺寸之间的关系

设置卷积核大小(kernel_size)卷积步长 (stride),特征图**填充宽度 (padding)**等参数

原文链接:https://zhuanlan.zhihu.com/p/163017446

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,我理解了你的问题。以下是一个简单的实现,其中包含一个名为`MultiChannelConv2D`的类,用于执行多通道卷积神经网络操作,具有`input_image`、`out_channels`、`stride`、`kernel_size`和`padding`等参数,并且包含一个`forward`方法,该方法返回卷积后的特征。 ```python import numpy as np class MultiChannelConv2D: def __init__(self, input_image, out_channels, stride, kernel_size, padding): self.input_image = input_image self.out_channels = out_channels self.stride = stride self.kernel_size = kernel_size self.padding = padding def forward(self): # 获取输入像的形状 n, c, h, w = self.input_image.shape # 计算输出像的形状 out_h = int((h + 2 * self.padding - self.kernel_size) / self.stride + 1) out_w = int((w + 2 * self.padding - self.kernel_size) / self.stride + 1) # 初始化输出特征 output = np.zeros((n, self.out_channels, out_h, out_w)) # 对每个输入通道和输出通道执行卷积操作 for i in range(self.out_channels): for j in range(c): # 初始化卷积kernel = np.random.randn(self.kernel_size, self.kernel_size) # 对输入像进行填充 padded_input = np.pad(self.input_image[:, j, :, :], [(0, 0), (self.padding, self.padding), (self.padding, self.padding)], mode='constant') # 对输入像执行卷积操作 for k in range(out_h): for l in range(out_w): output[:, i, k, l] += np.sum(kernel * padded_input[:, k*self.stride:k*self.stride+self.kernel_size, l*self.stride:l*self.stride+self.kernel_size], axis=(1, 2)) # 返回输出特征 return output ``` 在这个实现中,我们使用`numpy`来处理像数据和卷积核,`input_image`参数应该是一个四维张量,其形状为`(n, c, h, w)`,其中`n`是批量大小,`c`是通道数,`h`和`w`是像的高度和宽度。`out_channels`参数指定了卷积操作的输出通道数,`stride`参数指定了卷积操作的步幅,`kernel_size`参数指定了卷积核的大小,`padding`参数指定了输入像的填充大小。 在`forward`方法中,我们首先计算输出特征的形状,然后初始化输出特征。接下来,我们对每个输入通道和输出通道执行卷积操作。对于每个输出通道,我们在每个输入通道上使用不同的卷积核进行卷积操作,并对每个像素点进行求和以获得输出特征中的值。最后,我们返回输出特征。 需要注意的是,这只是一个简单的实现,实际上在卷积神经网络中还有许多其他的操作和层,如池化层、激活函数、批量标准化等等。但是,这个简单的实现足以说明卷积神经网络的基本原理和实现方法。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值