megengine.data.transform.Compose¶
-
class
Compose(transforms=[], batch_compose=False, shuffle_indices=None, *, order=None)[源代码]¶ 基类：
megengine.data.transform.vision.transform.VisionTransform将若干变换组合在一起。
- 参数
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)] )
方法
__init__([transforms, batch_compose, …])初始化方法。
apply(input)对单个输入数据实施数据增强。
apply_batch(inputs)对一批（batch）的输入数据实施数据增广。