ReflectionPad2d() 类似于一种 镜像填充
说说公式,及属性 具体官方文档很详细 不过不够通俗= =
官方文档上 是这个
表示第一眼 有点不理解= =
但是输出是
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.youkuaiyun.com/bc521bc/article/details/85134539