Pad¶

参数¶

• padding (int|list|tuple) - 在图像边界上进行填充的范围。如果提供的是单个 int 值，则该值用于填充图像所有边；如果提供的是长度为 2 的元组/列表，则分别为图像左/右和顶部/底部进行填充；如果提供的是长度为 4 的元组/列表，则按照左，上，右和下的顺序为图像填充。

• fill (int|list|tuple，可选) - 用于填充的像素值。仅当 padding_mode 为 constant 时参数值有效。 默认值：0。 如果参数值是一个长度为 3 的元组，则会分别用于填充 R，G，B 通道。

• padding_mode (string，可选) - 填充模式。支持: constant, edge, reflect 或 symmetric。 默认值：constant。

  • constant 表示使用常量值进行填充，该值由 fill 参数指定；

  • edge 表示使用图像边缘像素值进行填充；

  • reflect 表示使用原图像的镜像值进行填充（不使用边缘上的值）；比如：使用该模式对 [1, 2, 3, 4] 的两端分别填充 2 个值，结果是 [3, 2, 1, 2, 3, 4, 3, 2]。

  • symmetric 表示使用原图像的镜像值进行填充（使用边缘上的值）；比如：使用该模式对 [1, 2, 3, 4] 的两端分别填充 2 个值，结果是 [2, 1, 1, 2, 3, 4, 4, 3]。

• keys (list[str]|tuple[str]，可选) - 与 BaseTransform 定义一致。默认值: None。

形状¶

• img (PIL.Image|np.ndarray|Paddle.Tensor) - 输入的图像数据，数据格式为'HWC'。

• output (PIL.Image|np.ndarray|Paddle.Tensor) - 返回填充后的图像数据。

返回¶

计算 Pad 的可调用对象。

代码示例¶

import numpy as np
from PIL import Image
from paddle.vision.transforms import Pad

transform = Pad(2)

fake_img = Image.fromarray((np.random.rand(224, 224, 3) * 255.).astype(np.uint8))

fake_img = transform(fake_img)
print(fake_img.size)