卷积层、全连接层的FLOPs和参数量计算

卷积层

假设输入的feature map的维度(N, C, H, W),卷积层的的维度为(F, C, HH, WW),输出feature map的维度为(N, F, H', W'),那么不考虑样本数量

其中,括号内第一项是卷积的乘法计算量,第二项是卷积的加法计算量,当有偏置时存在+1。

参数量:

卷积层、全连接层的FLOPs和参数量计算_第1张图片

全连接层

假设输入的数据维度(N, D),隐藏层维度(D, out),那么

参数量:

你可能感兴趣的