mindspore.dataset.vision.py_transforms.HWC2CHW

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

Transpose the input numpy.ndarray image of shape (H, W, C) to (C, H, W).

Raises

TypeError – If the input is not numpy.ndarray.
TypeError – If the dimension of input is not 3.

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")