CNN中卷积层参数量与输出特征图(feature map)尺寸的计算公式

  1. 卷积层输入特征图(input feature map)的尺寸为: H i n p u t × W i n p u t × C i n p u t
    1. H i n p u t 表示输入特征图的高
    2. W i n p u t 表示输入特征图的宽
    3. C i n p u t 表示输入特征图的通道数(如果是第一个卷积层则是输入图像的通道数,如果是中间的卷积层则是上一层的输出通道数)
  2. 卷积层的参数有如下几个:
    1. 输出通道数为 K
    2. 正方形卷积核的边长为 F
    3. 步幅(stride)为 S
    4. 补零的行数和列数(padding)为 P
  3. 输出特征图(output feature map)的尺寸为 H o u t p u t × W o u t p u t × C o u t p u t ,其中每一个变量的计算方式如下:
    1. H i n p u t = ( H o u t p u t F + 2 P ) / S + 1
    2. W o u t p u t = ( W i n p u t F + 2 P ) / S + 1
    3. C o u t p u t = K
  4. 参数量大小的计算,分为weights和biases:
    1. 首先来计算weights的参数量: F × F × C i n p u t × K
    2. 接着计算biases的参数量: K
    3. 所以总参数量为: F × F × C i n p u t × K + K

猜你喜欢

转载自blog.csdn.net/gaishi_hero/article/details/81512404