mindspore.dataset.vision.ToType
- class mindspore.dataset.vision.ToType(data_type)[source]
Cast the input to a given MindSpore data type or NumPy data type.
It is the same as that of
mindspore.dataset.transforms.TypeCast
.Note
This operation is executed on the CPU by default, but it is also supported to be executed on the GPU or Ascend via heterogeneous acceleration.
- Parameters
data_type (Union[mindspore.dtype, numpy.dtype]) – The desired data type of the output image, such as
numpy.float32
.- Raises
TypeError – If data_type is not of type
mindspore.dtype
ornumpy.dtype
.
- Supported Platforms:
CPU
GPU
Ascend
Examples
>>> import mindspore.dataset as ds >>> import mindspore.dataset.vision as vision >>> import numpy as np >>> from mindspore.dataset.transforms import Compose >>> >>> transforms_list = Compose([vision.Decode(to_pil=True), ... vision.RandomHorizontalFlip(0.5), ... vision.ToTensor(), ... vision.ToType(np.float32)]) >>> # apply the transform to dataset through map function >>> image_folder_dataset = ds.ImageFolderDataset("/path/to/image_folder_dataset_directory") >>> image_folder_dataset = image_folder_dataset.map(operations=transforms_list, ... input_columns="image")
- Tutorial Examples: