mindspore.dataset.vision.ToType

class mindspore.dataset.vision.ToType(data_type)[源代码]

将输入转换为指定的MindSpore数据类型或NumPy数据类型。

效果同 mindspore.dataset.transforms.TypeCast 。

说明

此操作默认通过 CPU 执行，也支持异构加速到 GPU 或 Ascend 上执行。

参数：

data_type (Union[mindspore.dtype, numpy.dtype]) - 输出图像的数据类型，例如 numpy.float32 。

异常：

TypeError - 当 data_type 的类型不为 mindspore.dtype 或 numpy.dtype 。

支持平台：

CPU GPU Ascend

样例：

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

教程样例：

视觉变换样例库