nn.ReflectionPad2d（镜像填充）

Posted 2021-03-13 lyp1010

1、官方文档上 是这个     

 

 

2、输出是   

H(out) = H(in) + paddingTop + paddingBottom
W(out) = W(in) + paddingLeft + paddingRight 
直接来例子试试    最重要的是下面 # laft, right, top, bottom  这个注释

input = torch.randn(64, 3, 220, 220) # input size

# 输入4-tuple
pad = nn.ReflectionPad2d((3, 3, 5, 5))     # laft, right, top, bottom
output = pad(input) # size(64, 3, 230, 226)
# 输入int
pad = nn.ReflectionPad2d(3)
output = pad(input) # size(64, 3, 226, 226)
通过运算 发现的确是这样的结果      所以记住公式就行了

转自：https://blog.csdn.net/bc521bc/article/details/85134539