mindspore.dataset.vision.Decode

查看源文件
class mindspore.dataset.vision.Decode(to_pil=False)[源代码]

将输入的压缩图像解码为RGB格式。当前支持的图片类型:JPEG、BMP、PNG、TIFF、GIF(需要指定 to_pil=True)、WEBP(需要指定 to_pil=True)。

参数:
  • to_pil (bool,可选) - 是否将图像解码为PIL数据类型。若为 True ,图像将被解码为PIL数据类型,否则解码为NumPy数据类型。默认值: False

异常:
  • RuntimeError - 如果输入图像不是一维序列。

  • RuntimeError - 如果输入数据不是合法的图像字节数据。

  • RuntimeError - 如果输入数据已经是解码的图像数据。

支持平台:

CPU Ascend

样例:

>>> import mindspore.dataset as ds
>>> import mindspore.dataset.vision as vision
>>>
>>> # Eager usage
>>> import numpy as np
>>> raw_image = np.fromfile("/path/to/image/file", np.uint8)
>>> decoded_image = vision.Decode()(raw_image)
>>>
>>> # Pipeline usage
>>> image_folder_dataset = ds.ImageFolderDataset("/path/to/image_folder_dataset_directory")
>>> transforms_list = [vision.Decode(), vision.RandomHorizontalFlip()]
>>> image_folder_dataset = image_folder_dataset.map(operations=transforms_list,
...                                                 input_columns=["image"])
教程样例:
device(device_target='CPU')[源代码]

指定该变换执行的设备。

  • 当执行设备是 Ascend 时,输入数据仅支持 uint8 类型。

参数:
  • device_target (str, 可选) - 算子将在指定的设备上运行。当前支持 CPU 。默认值: CPU

异常:
  • TypeError - 当 device_target 的类型不为str。

  • ValueError - 当 device_target 的取值不为 CPU

支持平台:

CPU

样例:

>>> import mindspore.dataset as ds
>>> import mindspore.dataset.vision as vision
>>> from mindspore.dataset.vision import Inter
>>>
>>> decode_op = vision.Decode().device("Ascend")
>>> resize_op = vision.Resize([100, 75], Inter.BICUBIC)
>>> transforms_list = [decode_op, resize_op]
>>> image_folder_dataset = ds.ImageFolderDataset("/path/to/image_folder_dataset_directory")
>>> image_folder_dataset = image_folder_dataset.map(operations=transforms_list,
...                                                 input_columns=["image"])