Advertisement

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)

还没有任何评论哟~