如何计算卷积层的尺寸、参数量和浮点操作数(FLOPs)

1. 卷积的计算过程

 

- 输入特征图(Hi,Wi,Ci)，采用卷积核(K,K,Ci, p, s)，其中K为卷积核大小，Ci既是输入特征的通道数(深度)也是卷积核的通道数(深度)，也就是说卷积核的通道数要跟特征图的通道数匹配才能进行每一次卷积，单通道灰度图对应单通道卷积核，三通道彩色图对应三通道卷积核。

- 输出特征图(Ho,Wo,Co)中Co的计算方式：Co可以随便定义，但每一层输出特征图对应一个独立的卷积核进行一轮独立的卷积操作即Co个数一方面决定了输出特征图的通道数，另一方面也决定了卷积核的个数，要进行几轮卷积就生成Co张特征图，这也是卷积操作能够调整特征图深度的原因。

- 输出特征图(Ho,Wo,Co)中Ho, Wo的计算方式：取决于卷积核参数(3x3xCi, p, s)中的p和s，其中p代表padding边缘填充，s代表stride步长。计算公式如下，其实很好理解，就是把输入特征图减掉卷积核大小加两倍padding，然后除以步长：

Wo = (Wi - K + 2p)/s + 1