Pad¶

class torchvision.transforms.v2.Pad(padding: Union[int, Sequence[int]], fill: Union[int, float, Sequence[int], Sequence[float], None, dict[Union[type, str], Union[int, float, collections.abc.Sequence[int], collections.abc.Sequence[float], NoneType]] = 0, padding_mode: Literal['constant', 'edge', 'reflect', 'symmetric'] = 'constant')[源代码]¶

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

如果输入是 torch.Tensor 或 TVTensor（例如 Image、Video、BoundingBoxes 等），它可以具有任意数量的前导批次维度。例如，图像可以具有 [..., C, H, W] 形状。边界框可以具有 [..., 4] 形状。

参数:

padding (int 或序列) –
每个边缘的填充。如果提供单个整数，则将其用于填充所有边。如果提供长度为 2 的序列，则分别表示左/右和顶部/底部的填充。如果提供长度为 4 的序列，则分别表示左、上、右和底部边缘的填充。

注意

在 torchscript 模式下，不支持将 padding 作为单个 int，请使用长度为 1 的序列：[padding, ]。
fill (数字或元组或字典, 可选) – 当 padding_mode 为 constant 时使用的像素填充值。默认为 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：通过反射图像进行填充，但不重复边缘处的最后一个值。例如，在 reflect 模式下，将 [1, 2, 3, 4] 的两端各填充 2 个元素将导致 [3, 2, 1, 2, 3, 4, 3, 2]
- symmetric：通过反射图像进行填充，重复边缘处的最后一个值。例如，在 symmetric 模式下，将 [1, 2, 3, 4] 的两端各填充 2 个元素将导致 [2, 1, 1, 2, 3, 4, 4, 3]