主要就是两个公式:
1、nn.Conv2d
h/w = (h/w - kennel_size + 2padding) / stride + 1
详细可参考:nn.Conv2d卷积
2、nn.MaxPool2d
out_size=((input_size-kernel_size+2padding)/stride) + 1
主要就是两个公式:
1、nn.Conv2d
h/w = (h/w - kennel_size + 2padding) / stride + 1
详细可参考:nn.Conv2d卷积
2、nn.MaxPool2d
out_size=((input_size-kernel_size+2padding)/stride) + 1