CNN VGG 参数、输入输出等计算

卷积层feature map大小计算:

卷积层输出feature map大小计算公式: [公式] (一般向下取整)。

普通卷积层参数计算:

[公式]

卷积层参数由Filter计算,参数量=(filter size * 前一层特征图的通道数 )* 当前层filter数量

VGG-16为例,Conv1-1,输入 [公式] ,64个 [公式] filter,输出feature map [公式]

Conv1-1的参数量为 [公式]

Conv2-1,输入 [公式] ,128个 [公式] filter,输出feature map [公式]

Conv2-1的参数量 [公式]

全连接层的参数量:

上面已经说过卷积层的参数量计算方法了,那如何计算全连接层的参数量?其实和卷积层参数量的计算方法是一样的

VGG-16最后一次卷积得到的feature map为 [公式] ,全连接层是将feature map展开成一维向量 [公式]实际上,我们就是用4096个[公式] 的filter去做卷积(可以理解为是一个卷积层)。

我们就可以计算第一个FC的参数量 [公式]

一次卷积的计算量,以VGG-16为例,Conv1-1,输入 [公式] ,64个 [公式] filter,输出feature map [公式]

feature map中的每一个像素点,都是64个 [公式] filter 共同作用于原图计算一次得到的,所以它的计算量为 [公式]

已经知道单个像素的计算量,那乘以feature map所有像素,就是一次卷积的计算量: [公式]

 

posted @ 2022-04-26 11:26  ka2uha  阅读(650)  评论(0编辑  收藏  举报