快捷方式

填充

class torchvision.transforms.v2.Pad(padding: Union[int, Sequence[int]], fill: Union[int, float, Sequence[int], Sequence[float], None, Dict[Union[Type, str], Optional[Union[int, float, Sequence[int], Sequence[float]]]] = 0, padding_mode: Literal['constant', 'edge', 'reflect', 'symmetric'] = 'constant')[source]

用给定的“填充”值在所有边上填充输入。

如果输入是 torch.TensorTVTensor(例如 ImageVideoBoundingBoxes 等),它可以具有任意数量的领先批次维度。例如,图像可以具有 [..., C, H, W] 形状。边界框可以具有 [..., 4] 形状。

参数:
  • padding (int序列) –

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

    注意

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

  • 填充 (数字元组字典, 可选) – 当 padding_mode 为常数时使用的像素填充值。默认值为 0。如果是一个长度为 3 的元组,则分别用于填充 R、G、B 通道。填充值也可以是将数据类型映射到填充值的字典,例如 fill={tv_tensors.Image: 127, tv_tensors.Mask: 0},其中 Image 将填充 127,而 Mask 将填充 0。

  • padding_mode (字符串, 可选) –

    填充类型。应为:constant、edge、reflect 或 symmetric。默认值为“constant”。

    • constant: 用常数值填充,该值由 fill 指定

    • edge: 用图像边缘的最后一个值填充。

    • reflect: 用图像的反射填充,但不重复边缘上的最后一个值。例如,用反射模式在两侧填充 2 个元素的 [1, 2, 3, 4] 将得到 [3, 2, 1, 2, 3, 4, 3, 2]

    • symmetric: 用图像的反射填充,重复边缘上的最后一个值。例如,用对称模式在两侧填充 2 个元素的 [1, 2, 3, 4] 将得到 [2, 1, 1, 2, 3, 4, 4, 3]

使用 Pad 的示例

变换的说明

变换的说明

文档

访问 PyTorch 的全面的开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源