在设计以及实现模型时,计算每层卷积输出特征图大小是很重要的,学会计算卷积操作的计算量与参数量,可以让我们评估对比不同模型。
输出特征图大小
output_size = (input_size-kernel_size+2*padding)//stride+1。(//表示取整,也就是向下取整)
卷积操作的计算量
卷积操作的参数量
parameters = in_channels * kernel_size_w * kernel_size_h * out_channels + bias。(其中bias=out_channels)。