快捷方式

填充

class torchvision.transforms.Pad(padding, fill=0, padding_mode='constant')[source]

在给定图像的所有侧使用给定的“填充”值进行填充。如果图像为 torch 张量,则预期其形状为 […, H, W],其中 … 表示对于模式 reflect 和 symmetric 最多 2 个前导维度,对于模式 edge 最多 3 个前导维度,以及对于模式 constant 任意数量的前导维度

参数:
  • padding (int序列) –

    每个边界的填充。如果提供单个 int,则将其用于填充所有边界。如果提供长度为 2 的序列,则分别为左右和上下填充。如果提供长度为 4 的序列,则分别为左右、上、右和下边界的填充。

    注意

    在 torchscript 模式下,不支持将填充作为单个 int,请使用长度为 1 的序列: [padding, ]

  • fill (数字元组) – 常量填充的像素填充值。默认为 0。如果为长度为 3 的元组,则分别用于填充 R、G、B 通道。此值仅在 padding_mode 为常量时使用。对于 torch 张量,仅支持数字。对于 PIL 图像,仅支持 int 或元组值。

  • padding_mode (str) –

    填充类型。应该是:constant、edge、reflect 或 symmetric。默认为 constant。

    • constant:用常量值填充,此值由 fill 指定

    • edge:用图像边缘的最后一个值填充。如果输入 5D torch 张量,则将填充最后 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 的示例

变换图示

变换图示
forward(img)[source]
参数:

img (PIL 图像张量) – 要填充的图像。

返回值:

填充后的图像。

返回类型:

PIL 图像或张量

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源