mindspore.dataset.vision.py_transforms.ToTensor

class mindspore.dataset.vision.py_transforms.ToTensor(output_type=np.float32)

将输入的PIL或numpy.ndarray图像转换为指定数据类型的numpy.ndarray图像，此时像素值取值将由[0, 255]变为[0.0, 1.0]，图像的shape将由(H, W, C)变为(C, H, W)。

Note

输入图像中的像素值将从 [0, 255] 缩放为 [0.0, 1.0]。图像的数据类型将被转换为由 output_type 参数指定的类型，图像通道数保持不变。

参数：

• output_type (numpy.dtype，可选) - 输出图像的数据类型，默认值：numpy.float32。

异常：

• TypeError - 当输入图像的类型不为 PIL.Image.Image 或 numpy.ndarray 。

• TypeError - 输入图像的维度不为2或3。

支持平台：

CPU

样例：

>>> from mindspore.dataset.transforms.py_transforms import Compose
>>>
>>> # create a list of transformations to be applied to the "image" column of each data row
>>> transforms_list = Compose([py_vision.Decode(),
...                            py_vision.RandomHorizontalFlip(0.5),
...                            py_vision.ToTensor()])
>>> # apply the transform to dataset through map function
>>> image_folder_dataset = image_folder_dataset.map(operations=transforms_list,
...                                                 input_columns="image")