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

>>> import mindspore.dataset.vision.py_transforms as py_vision
>>> from mindspore.dataset.transforms.py_transforms import Compose
>>> import numpy as np
>>>
>>> Compose([py_vision.Decode(), py_vision.RandomHorizontalFlip(0.5),
>>>          py_vision.ToTensor(),
>>>          py_vision.ToType(np.float32)])