超详细的卷积后大小的计算公式 您所在的位置:网站首页 卷筒尺寸计算 超详细的卷积后大小的计算公式

超详细的卷积后大小的计算公式

2023-11-06 22:03| 来源: 网络整理| 查看: 265

计算公式定义

定义几个参数

输入图片大小 W×W 卷积核大小 F×F 步长 S padding的像素数 P 于是我们可以得出计算公式为: N = (W − F + 2P )/S+1

输出图片大小为 N×N 以resnet50为例,输入为[1,3,224,224],其中1为batchsize,3为通道数,224为height和width。

经过第一层卷积后,其大小为[1,64,112,112]

in_channels= 3//输入通道 out_channels= 64 //输出通道 nn.Conv2d(in_channels, out_channels, kernel_size=7, stride=2, padding=3, bias=False)

N = (W − F + 2P )/S+1 = (224-7+2x3)/2 + 1 = 112

解释

1为batchsize,不改变。 对于通道数,会生成与设定的输出通道个数相同个数的卷积核,对图片进行卷积,即卷积核的个数等于输出特征图的通道数。 得到最终输出大小为[1,64,112,112] (W − F + 2P )相当于计算除了第一次卷积后剩下的可用来卷积的大小 (W − F + 2P )/S为按照S大小的步长在刚刚得到的大小上可以向后移动多少次,即还可以做几次卷积 因为不包括第一次卷积,所以再加上一个1, 即N = (W − F + 2P )/S+1 输出大小 = (图片宽或高 - 卷积核大小 + padding大小)/ 步长 + 1 对于宽和高不同的图片可分别用上述公式计算,得到最终的输出大小。

卷积动态图解参考: https://cs231n.github.io/assets/conv-demo/index.html



【本文地址】

公司简介

联系我们

今日新闻

    推荐新闻

    专题文章
      CopyRight 2018-2019 实验室设备网 版权所有