Caffe中的卷积实现

来源:互联网 发布:施耐德和公牛 知乎 编辑:程序博客网 时间:2024/04/27 18:20

图像卷积的实现

矩阵相乘

卷积是乘累加运算,把图像和卷积核展开成列,可用矩阵相乘加速运算。比如,输入feature map是32*32*3,卷积核是5*5*3*20,padding是1,stride是1,则展开后得到的feature map矩阵宽度是5*5*3,高度是((32+2*1-5)/1+1)*((32+2*1-5)/1+1),卷积核矩阵高度是5*5*3,宽度是20。

im2col的过程

0 0