mindspore.dataset.vision.ImageBatchFormat

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

Data format of images after batch operation.

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

  • ImageBatchFormat.NHWC: Stores data in the order of batch (N), height (H), width (W), and channels (C).

  • ImageBatchFormat.NCHW: Stores data in the order of batch (N), channels (C), height (H), and width (W).