mindspore.dataset.vision.py_transforms.ToType
- class mindspore.dataset.vision.py_transforms.ToType(output_type)[源代码]
将输入的numpy.ndarray图像转换为指定数据类型。
参数:
output_type (numpy.dtype) - 输出图像的数据类型,例如
numpy.float32
。
异常:
TypeError - 输入图像的类型不为
numpy.ndarray
。
- 支持平台:
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")