卷积神经网络的核心之一——卷积运算,常用到函数im2col。它将包含批数量的4维数据转换成2维数据,以适应展开的滤波器。
im2col 的代码如下:
def im2col(input_data, filter_h, filter_w, stride=1, pad=0):
"""
Parameters
----------
input_data : 由(数据量, 通道, 高, 长)的4维数组构成的输入数据
filter_h : 滤波器的高
filter_w : 滤波器的长
stride : 步幅
pad : 填充
Returns
-------
col : 2维数组
"""
N, C, H, W = input_data.shape
out_h = (H + 2*pad - filter_h)//stride + 1
out_w = (W + 2*pad - filter_w)//stride

本文介绍了卷积神经网络中的关键函数im2col,它将4维数据转换为2维,以便进行卷积运算。通过一个2通道66数据和2个22滤波器的例子,详细解释了卷积运算过程,并展示了im2col函数如何将数据排列以匹配滤波器。最后,阐述了如何将4维数据转化为2维数据,以便进行进一步的运算。

2012

被折叠的 条评论
为什么被折叠?



