mindspore.dataset.vision.py_transforms.HWC2CHW

class mindspore.dataset.vision.py_transforms.HWC2CHW[source]

Transpose a NumPy image array; shape (H, W, C) to shape (C, H, W).

Examples

>>> from mindspore.dataset.transforms.py_transforms import Compose
>>> transforms_list = Compose([py_vision.Decode(),
...                            py_vision.HWC2CHW()])
>>> # apply the transform to dataset through map function
>>> image_folder_dataset = image_folder_dataset.map(operations=transforms_list,
...                                                 input_columns="image")