对theano中conv2d函数的理解

前言

因为最近在自学CNN,在认识了卷积的一些基本概念后,对卷积过程中的一些过程有困惑,故想借助theano的conv2d函数加深理解。

问题

首先,对于conv2d的输入,第一层的输入往往是[多少张图片,RGB通道数,图片有多少行,图片有多少列], 权重的形状往往是[卷积核数量,三层特征图数,卷积核宽,卷积核高]。

假设输入图片是[1, 2, 1, 5],权重是[2,2,1,2],那么卷积的结果是[1, 2, 1, 4]。

明明权重是2*2个1*2,为什么最后得到的特征图数会更前面的卷积核数相同?

其实我们说的卷积核一般都是三维的,它是[上一层特征图数,宽,高]。进行卷积运算后还有一个求和的运算,这才是完整的conv2d。

实验

测试代码如下:

inputs = T.tensor4(name='input', dtype='float64')

w_shp = (2, 1, 1, 2)
W = theano.shared(
    np.asarray(
        [
            [[[1., 1.]],
             [[1., 1.]]],
            [[[2., 2.]],
             [[1., 1.]]]
        ],
        dtype=inputs.dtype),
    name='W')

conv_out = conv2d(inputs, W)

f = theano.function([inputs], conv_out)

i = np.asarray([
    [[[1., 2., 3., 4., 5.]],
     [[1., 2., 3., 4., 5.]]]
], dtype='float64')
ii = f(i)
print(i.shape)
print(W.get_value().shape)
print(ii)
print(ii.shape)

输出的结果如下:
这里写图片描述

输入为(1,2,3,4,5),(1,2,3,4,5)
对于第一个卷积核(1,1)(1,1)得到的结果为(3,5,7,9),(3,5,7,9),两个相加就是第一个输出(6,10,14,18)。

总结

刚开始学,没有深入了解细节很多时候都会走弯路。特别是刚开始是用的是MNIST的单通道的进行学习,一旦用三通道的就开始傻了。

但是既然该躺的坑都躺过了,后面就没有什么可怕的了。

你可能感兴趣的:(Theano)