计算padding

计算padding
pytorch拿别人的网络套进自己的数据集上，不知道输入数据大小的时候可以反推一下。

借鉴博客: Pytorch 卷积层的padding计算

# 1. in_channels=1, out_channels=32, kernel_size=11, stride=1, padding=5
W = 205
F = 11
P = 5
S = 1

def calculate_n(w, f, p, s):
    n = (w - f + 2 * p) / s + 1
    return n


def calculate_p(n, w, f, s):
    p = ((n - 1) * s - w + f) / 2
    return p


n_size = calculate_n(W, F, P, S)
print(n_size)

padding = calculate_p(n_size, W, F, S)
print(padding)