mindspore.dataset.vision.write_png

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

将图像数据保存为PNG文件。

参数：

  • filename (str) - 要写入的文件的路径。

  • image (Union[numpy.ndarray, mindspore.Tensor]) - 要写入的图像数据。

  • compression_level (int, 可选) - 生成PNG文件的压缩级别，取值范围为[0, 9]。默认值： 6。

异常：

  • TypeError - 如果 filename 不是str类型。

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

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

  • RuntimeError - 如果 filename 不存在或不是普通文件。

  • 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)
>>> vision.write_png("/path/to/file", image)