mindspore.dataset.vision.encode_png

查看源文件
mindspore.dataset.vision.encode_png(image, compression_level=6)[源代码]

将输入的图像编码为PNG数据。

参数:
  • image (Union[numpy.ndarray, mindspore.Tensor]) - 编码的图像。

  • compression_level (int, 可选) - 编码压缩因子,取值范围为[0, 9]。默认值: 6

返回:
  • numpy.ndarray, 一维uint8类型数据。

异常:
  • TypeError - 如果 image 不是numpy.ndarray或mindspore.Tensor类型。

  • TypeError - 如果 compression_level 不是int类型。

  • RuntimeError - 如果 image 的数据类型不是uint8类型。

  • RuntimeError - 如果 image 的shape不是 <H, W> 或 <H, W, 1> 或 <H, W, 3>。

  • RuntimeError - 如果 compression_level 小于0或大于9。

支持平台:

CPU

样例:

>>> import mindspore.dataset.vision as vision
>>> import numpy as np
>>> # Generate a random image with height=120, width=340, channels=3
>>> image = np.random.randint(256, size=(120, 340, 3), dtype=np.uint8)
>>> png_data = vision.encode_png(image)