Pad¶
- class torchvision.transforms.Pad(padding, fill=0, padding_mode='constant')[source]¶
使用给定的“pad”值在图像的所有边上进行填充。如果图像是 torch Tensor,则期望其形状为 […, H, W],其中 … 表示模式 reflect 和 symmetric 最多 2 个前导维度,模式 edge 最多 3 个前导维度,模式 constant 则为任意数量的前导维度
- 参数:
padding (int 或 sequence) –
每个边框上的填充。如果提供单个 int,则用于填充所有边框。如果提供长度为 2 的序列,则分别为左右和上下的填充。如果提供长度为 4 的序列,则分别为左、上、右和下边框的填充。
注意
在 torchscript 模式下,不支持将 padding 作为单个 int,请使用长度为 1 的序列:
[padding, ]
。fill (number 或 tuple) – 用于常量填充的像素填充值。默认为 0。如果为长度为 3 的元组,则分别用于填充 R、G、B 通道。此值仅在 padding_mode 为 constant 时使用。torch Tensor 仅支持数字。PIL Image 仅支持 int 或 tuple 值。
padding_mode (str) –
填充类型。应为:constant、edge、reflect 或 symmetric。默认为 constant。
constant:用常量值填充,此值由 fill 指定
edge:用图像边缘的最后一个值填充。如果输入 5D torch Tensor,则将填充最后 3 个维度而不是最后 2 个维度
reflect:用图像的反射填充,而不重复边缘上的最后一个值。例如,在 reflect 模式下,用 2 个元素在两侧填充 [1, 2, 3, 4] 将得到 [3, 2, 1, 2, 3, 4, 3, 2]
symmetric:用图像的反射填充,并重复边缘上的最后一个值。例如,在 symmetric 模式下,用 2 个元素在两侧填充 [1, 2, 3, 4] 将得到 [2, 1, 1, 2, 3, 4, 4, 3]
使用
Pad
的示例