mindspore.dataset.vision.py_transforms.ToType

class mindspore.dataset.vision.py_transforms.ToType(output_type)[源代码]

将输入的numpy.ndarray图像转换为指定数据类型。

参数:

  • output_type (numpy.dtype) - 输出图像的数据类型,例如 numpy.float32

异常:

支持平台:

CPU

样例:

>>> import numpy as np
>>> from mindspore.dataset.transforms.py_transforms import Compose
>>>
>>> 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")