megengine.data.transform.ToMode¶
- class ToMode(mode='CHW', *, order=None)[源代码]¶
将输入数据变换到目标模式。例如，大多数变换使用HWC模式图像，而神经网络可能使用CHW模式的输入张量。
- 参数
mode – 输入的输出模式。默认情况下使用”CHW”模式。
order – 与
VisionTransform一致
方法
apply(input)对单个输入数据实施数据增强。
apply_batch(inputs)对一批（batch）的输入数据实施数据增广。