megengine.data.transform.Compose

class Compose(transforms=[], batch_compose=False, shuffle_indices=None, *, order=None)[源代码]

将若干变换组合在一起。

参数
  • transforms – 待组合的 VisionTransform 列表。

  • batch_compose – 是否对数据增广方法进行打乱。如果为True,使用原始的数据增广顺序。否则,按照shuffle_indices中给定的顺序对数据增广方法进行打乱。

  • shuffle_indices – 用于随机排序(random shuffle)的索引,从1开始。例如,若shuffle_indices 为[(1, 3), (2, 4)],则序列中的第一个数据增广方法和第三个数据增广方法有可能进行shuffle,同时第二和第四个有可能进行打乱。

  • order – 与 VisionTransform 一致

实际案例

from megengine.data.transform import RandomHorizontalFlip, RandomVerticalFlip, CenterCrop, ToMode, Compose

transform_func = Compose([
RandomHorizontalFlip(),
RandomVerticalFlip(),
CenterCrop(100),
ToMode("CHW"),
],
shuffle_indices=[(1, 2, 3)]
)

方法

apply(input)

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

apply_batch(inputs)

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