IDL-im2col和col2im方法
发布时间
阅读量:
阅读量
IDL im2col与col2im函数应用示例
在卷积神经网络中,针对卷积层与池化层所涉及的im2col以及反向传播过程中的col2im方法,本人获得了一些灵感。考虑到IDL语言本身并未提供相应的im2col和col2im函数,因此决定自行开发相关的IDL函数。接下来,将直接展示实现代码。
1. im2col
;input_data: 由(高, 长)的2维数组构成的输入数据
;filter_h: 滤波器的高
;filter_w: 滤波器的长
;stride=stride: 步幅
;pad=pad: 填充的个数(默认填充值为0)
;return 二维数组
FUNCTION fun_im2col, input_data, filter_h, filter_w, stride=stride, pad=pad
param_number = N_PARAMS() ;返回非关键字参数的个数(即位置参数的个数)
IF param_number EQ 3 THEN BEGIN
arr_info = SIZE( input_data )
w = arr_info[1] ;w和h为矩阵的长(列)和高(行)
h = arr_info[2]
全部评论 (0)
还没有任何评论哟~
