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)
通过运算 发现的确是这样的结果 所以记住公式就行了