ToMode

class ToMode(mode='CHW', *, order=None)[源代码]

将输入数据变换到目标模式。例如,大多数变换使用HWC模式图像,而神经网络可能使用CHW模式的输入张量。

参数
  • mode – 输入的输出模式。默认情况下使用”CHW”模式。

  • order – 与 VisionTransform 一致