在 Caffe 中如何计算卷积?

最好能图文并茂哈
关注者
900
被浏览
55612
【多图】

前面几楼都很详细了,正好今天在Berkeley给了关于CNN优化的talk,顺便把卷积的几张幻灯片贴上来,希望能有点用:
最后一页没画,但是基本上就是Filter Matrix乘以Feature Matrix的转置,得到输出矩阵Cout x (H x W),就可以解释为输出的三维Blob(Cout x H x W)。

Caffe里用的是CHW的顺序,有些library也会用HWC的顺序(比如说CuDNN是两个都支持的),这个在数学上其实差别不是很大,还是一样的意思。

顺便广告一下我吐槽Caffe里面卷积算法的链接:Convolution in Caffe: a memo · Yangqing/caffe Wiki · GitHub