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 或 序列) –
每个边界上的填充。如果提供单个整数，则此整数用于填充所有边界。如果提供长度为 2 的序列，则分别表示左/右和顶部/底部的填充。如果提供长度为 4 的序列，则分别表示左、上、右和底部的填充。
注意
在 torchscript 模式下，不支持将 padding 作为单个 int，请使用长度为 1 的序列：
[padding, ]。fill (数字或元组) – 常量填充的像素填充值。默认为 0。如果为长度为 3 的元组，则分别用于填充 R、G、B 通道。仅当 padding_mode 为 constant 时使用此值。对于 torch Tensor，仅支持数字。对于 PIL Image，仅支持整数或元组值。
padding_mode (str) –
填充类型。应为：constant、edge、reflect 或 symmetric。默认为 constant。
constant：用常数值填充，该值通过 fill 指定
edge：用图像边缘的最后一个值填充。如果输入是 5D torch Tensor，将填充最后 3 个维度而不是最后 2 个维度
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]
Pad的示例
