megengine.data.transform.Pad

class Pad(size=0, value=0, mask_value=0, *, order=None)[源代码]

填充输入数据。

参数
  • size – 输入图像的填充尺寸,它可以是单个整数或序列。如果它是一个整数,则在四个方向上填充输入图像。如果它是包含两个整数的序列,则仅填充图像的下侧和右侧。如果它是一个包含四个整数的序列,则按给定大小依次填充图像的上,下,左,右侧。

  • value – 填充图像的值,可以是一个整型或浮点型序列。如果为浮点值,图像的数据类型也会随之被转换为浮点。

  • mask_value – 填充分割图的值。

  • order – 与 VisionTransform 一致。

方法

apply(input)

对单个输入数据实施数据增强。

apply_batch(inputs)

对一批(batch)的输入数据实施数据增广。