mindspore.dataset.vision.ImageBatchFormat

class mindspore.dataset.vision.ImageBatchFormat[源代码]

图像批处理输出格式枚举类。

可选枚举值为:ImageBatchFormat.NHWC、ImageBatchFormat.NCHW。

  • ImageBatchFormat.NHWC - 按批次N、高度H、宽度W、通道C的顺序存储数据。

  • ImageBatchFormat.NCHW - 按批次N、通道C、高度H、宽度W的顺序存储数据。