mindspore.dataset.vision.ImageBatchFormat

View Source On Gitee
class mindspore.dataset.vision.ImageBatchFormat[source]

Data Format of images after batch operation.

Possible enumeration values are: ImageBatchFormat.NHWC, ImageBatchFormat.NCHW.

  • ImageBatchFormat.NHWC: in orders like, batch N, height H, width W, channels C to store the data.

  • ImageBatchFormat.NCHW: in orders like, batch N, channels C, height H, width W to store the data.