megengine.data.transform.TorchTransformCompose

class TorchTransformCompose(transforms, *, order=None)[源代码]

torchvision中变换的Compose类,仅支持PIL图像,但不支持某些torchvision中张量上的变换,比如Normalize和ToTensor。

参数
  • transforms – 与 Compose 一致。

  • order – 与 VisionTransform 一致。

方法

apply(input)

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

apply_batch(inputs)

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