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）的输入数据实施数据增广。

VisionTransform

TorchTransformCompose