Caffe中卷积的实现

方法:将卷积转化为矩阵相乘 两步: (1)调用im2col函数,将图像(feature map)转化为矩阵 (2)调用GEMM函数,做实际的运算 优点:逻辑简单,易于实现 缺点:耗费大量内存 feature map每一个卷积位置(卷积核将要滑过的位置),展开成一个 C∗K∗K 维的行向量 继续展开…… 按照这个方式展开完毕之后,得到一个矩阵: H∗W 行(因为有 H∗W 个卷积位置), C∗K∗K
相关文章
相关标签/搜索