mindspore.dataset.vision.ImageBatchFormat

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.

static to_c_type(image_batch_format)[source]

Function to return C type for ImageBatchFormat.