二维卷积输出特征图的计算公式

---

计算公式

首先附上计算公式：

参数说明

• H：高度；
• W：宽度；
• N：batch-size的大小；
• Cin：输入的通道数；
• Cout：卷积后输出的通道数；
• dilation：膨胀率，的默认值是1；
• stride：步长，默认是1。

---

说明：如果输入的是图像，一般高度H和宽度W相同，所以只需要算一个就好。

如果膨胀率 dilation=1，则公式为：

示例

输入一个 256 * 256 的向量，经过下面的三层卷积，套入公式计算特征图的大小：

nn.Sequential(
    nn.ReflectionPad2d(3),  
    nn.Conv2d(4, 64, 7),        
    nn.Conv2d(64, 128, 4, stride=2, padding=1),
    nn.Conv2d(128, 256, 4, stride=2, padding=1),
)
1
2
3
4
5
6

ReflectionPad2d(3)是以输入图像以最外围像素为对称轴，做四周的轴对称镜像填充,扩充3个单位。
以下是根据公式的计算过程：

可以看到，经过一次填充和三次卷积，大小变为了原来的1/4。