Caffe中卷积层的实现

作者：xg123321123

出处：http://blog.csdn.net/xg123321123/article/details/53319080

声明：版权所有，转载请联系作者并注明出处

1 简述

• 使用im2col分别将featrue maps和filter转换成矩阵；
• 调用GEMM（GEneralized Matrix Multiplication）对两矩阵内积，这样一来卷积操作就被转化为了矩阵乘法运算。
• 原理图如下
  

将尺寸为K×K的卷积核在某个位置对应的feature map区域表示为K×K的一维向量；
将feature map各个通道对应的向量之间，串联起来；
那么尺寸K×K的卷积核在某个位置对应的各个通道的feature map，组合起来就是长度为C×K×K的一维向量。

当卷积核对应到新的位置上，又得到新的一维向量。

那么卷积核对应整张图片的所有位置，就得到一个（H×W）×（C×K×K）的矩阵（假设卷积核每次移动的步数为1）。

同样地，将卷积核也表示为一维向量；
用于输出一个特征图的卷积核对应一个C×K×K的一维向量（因为用于输出同一个特征图的卷积核是共享权值的，所以在整张图的各个位置进行卷积的都是同一个卷积核）；
假设要输出$C_{out}$个特征图，就需要 Cout