mindspore.dataset.vision.py_transforms.ToType
- class mindspore.dataset.vision.py_transforms.ToType(output_type)[source]
Convert the input NumPy image array to desired NumPy dtype.
- Parameters
output_type (NumPy datatype) – The datatype of the NumPy output, e.g. numpy.float32.
Examples
>>> from mindspore.dataset.transforms.py_transforms import Compose >>> import numpy as np >>> transforms_list =Compose([py_vision.Decode(), ... py_vision.RandomHorizontalFlip(0.5), ... py_vision.ToTensor(), ... py_vision.ToType(np.float32)]) >>> # apply the transform to dataset through map function >>> image_folder_dataset = image_folder_dataset.map(operations=transforms_list, ... input_columns="image")