python将二值图保存为png

原创

mob64ca12e04e7a 2024-10-15 04:23:37 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e04e7a的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python将二值图保存为PNG格式

在数字图像处理中，二值图（Binary Image）是一种仅使用两种颜色（通常是黑与白）的图像。这种图像在模式识别、图像分割和图形处理等领域有着广泛的应用。本文将介绍如何使用Python将二值图保存为PNG格式。

需要的库

首先，我们需要安装一些必要的库，包括numpy和PIL（Pillow）。可以通过以下命令安装：

pip install numpy Pillow

二值图的生成

二值图像通常是由灰度图像或彩色图像处理得到的。我们可以使用numpy生成一个简单的二值图。例如，创建一个 10x10 的随机二值图像：

import numpy as np

# 生成随机二值数组
binary_array = np.random.choice([0, 1], size=(10, 10), p=[0.5, 0.5])  # 50%概率生成0和1

图像保存

生成二值图后，我们可以使用PIL库将其保存为PNG格式。以下是完整的代码示例：

from PIL import Image
import numpy as np

# 生成随机二值数组
binary_array = np.random.choice([0, 1], size=(10, 10), p=[0.5, 0.5])

# 将0和1转换为255和0，因为PIL需要以255代表白色（背景），以0代表黑色（前景）
binary_image = Image.fromarray(binary_array * 255).convert('L')

# 保存图像为PNG格式
binary_image.save('binary_image.png')

在上面的代码中，Image.fromarray将numpy的数组转换为PIL图像对象。convert('L')方法将图像转换为灰度模式，确保输出为二值图像。最后，调用save方法将图像保存为PNG格式。

代码解析

numpy.random.choice用于生成随机的二值数组。
Image.fromarray用于将numpy数组转换为图像对象。
save方法用于将图像文件保存到当前目录。

类图

为帮助更好地理解本示例中所使用的类和方法，下面是一个类图。

classDiagram
    class Image {
        +fromarray(array)
        +convert(mode)
        +save(filename)
    }
    class Random {
        +choice(a, size, p)
    }
    class Numpy {
        +random: Random
    }