计算公式
首先附上计算公式:
参数说明
- 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),
)
ReflectionPad2d(3)是以输入图像以最外围像素为对称轴,做四周的轴对称镜像填充,扩充3个单位。
以下是根据公式的计算过程:
可以看到,经过一次填充和三次卷积,大小变为了原来的1/4。