在卷积操作过程中,矩阵经卷积操作后的尺寸由以下几个因数决定:
- 输入图片大小 W×W
- Filter大小 F×F
- 步长 S
- padding的像素数 P
经卷积后的矩阵尺寸大小计算公式为:
N = (W − F + 2P ) / S + 1
但在实际应用中,有时会出现N为非整数的情况。
例如输入的矩阵 H=W=5,卷积核的F=2,S=2,Padding=1。
经计算我们得到的N =(5 - 2 + 2*1)/ 2 +1 = 3.5 ,此时在Pytorch中是如何处理呢?
在卷积过程中会直接将最后一行以及最后一列给忽略掉,以保证N为整数,此时N = (5 - 2 + 2*1 - 1)/ 2 + 1 = 3