nn.ReflectionPad2d()

最新推荐文章于 2024-11-07 09:18:03 发布

原创最新推荐文章于 2024-11-07 09:18:03 发布 · 1.6w 阅读

34 ·

CC 4.0 BY-SA版权

Pytorch 专栏收录该内容

17 篇文章

订阅专栏

本文深入解析了PyTorch中的ReflectionPad2d函数，详细介绍了其工作原理与使用方法。ReflectionPad2d是一种镜像填充方式，用于图像或信号处理中，通过反射边界元素来填充边界，以增加数据尺寸。文章通过实例演示了不同参数设置下ReflectionPad2d的输出变化，包括四元组和单一整数输入情况，展示了填充操作如何影响输出尺寸。

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)

通过运算发现的确是这样的结果所以记住公式就行了