如何使用Python快速去除图片背景并保存

你是否曾经遇到过这样的问题：你有一张图片，但是你只想保留其中的某些物体或图案，而将其他部分的背景删除？这可能是因为你需要将物体或图案添加到其他图片中，或者只是为了更好地突出显示这些物体。在这个教程中，我们将向你展示如何使用Python和PIL库快速去除图片的背景，并保存你想要的部分。

开篇故事

假设你是一名设计师，正在为一本儿童教育书籍制作插图。你已经有了一些美丽的插图，但它们都带有白色背景。为了使插图更好地融入书籍，你希望将背景去掉，只保留插图中的内容。手动去除每张图片的背景将非常耗时，所以你希望找到一种快速而有效的方法。

在这个背景下，你决定使用Python编写一个脚本，自动去除图片的背景，并将结果保存为具有透明背景的PNG图像。这将帮助你节省大量的时间，同时确保插图看起来更加专业。

准备工作

在开始之前，确保你已经安装了以下必要的库和工具：

• Python（建议使用3.x版本）
• Pillow（Python Imaging Library，用于处理图像）
• 一些包含带背景的图片

步骤 1：导入必要的库

首先，让我们导入所需的Python库。我们将使用Pillow库来处理图像。

from PIL import Image

步骤 2：编写背景去除函数

接下来，我们将编写一个函数，该函数将负责去除图片的背景并返回结果。我们将使用Pillow库中的convert方法来实现这一点。

def remove_background(image_path, save_path):
    img = Image.open(image_path)
    img = img.convert('RGBA')

    width, height = img.size

    for x in range(width):
        for y in range(height):
            r, g, b, a = img.getpixel((x, y))
            # 判断像素是否属于背景，这里假设背景是白色
            if r > 100 or g > 100 or b > 100:
                img.putpixel((x, y), (0, 0, 0, 0))
            else:
                img.putpixel((x, y), (r, g, b, 255))

    img.save(save_path)

这个函数接受输入图片的路径（image_path）和保存结果图片的路径（save_path）。它将图像转换为RGBA模式，然后遍历每个像素，将背景像素的Alpha通道设置为0，从而使其变为透明。我们假设背景是白色，但你可以根据需要修改这个条件。

步骤 3：使用函数去除背景

现在，我们可以使用刚刚编写的remove_background函数去除图片的背景。你可以将下面的代码添加到你的脚本中，用于处理单个图片：

input_image_path = 'input_image.png'  # 输入图片路径
output_image_path = 'output_image.png'  # 输出图片路径

remove_background(input_image_path, output_image_path)

步骤 4：批量处理多张图片

如果你有多张图片需要处理，你可以使用一个循环来批量处理它们。首先，将所有需要处理的图片放入一个文件夹中，然后使用以下代码来批量处理它们：

import os

input_folder = 'input_images'  # 包含输入图片的文件夹路径
output_folder = 'output_images'  # 包含输出图片的文件夹路径

if not os.path.exists(output_folder):
    os.makedirs(output_folder)

# 遍历文件夹中的所有图片文件
for filename in os.listdir(input_folder):
    if filename.endswith('.png') or filename.endswith('.jpg'):
        input_image_path = os.path.join(input_folder, filename)
        output_image_path = os.path.join(output_folder, filename)
        remove_background(input_image_path, output_image_path)

这段代码将遍历指定文件夹中的所有图片文件，并将它们的背景去除后保存到另一个文件夹中。

步骤 5：查看处理结果

你现在已经成功地去除了图片的背景，并保存了处理后的图片。你可以打开这些图片，查看它们的背景是否已经被成功去除，并且只剩下了前景物体或图案。

结语

通过使用Python和Pillow库，你可以轻松地去除图片的背景，并创建具有透明背景的PNG图像。这个技巧在许多情况下都非常有用，无论是用于设计、图像编辑还是其他创意项目。

无论你是一名设计师、开发者还是对图像处理感兴趣的人，都可以受益于这个简单而强大的背景去除技巧。现在，你可以将它应用到自己的项目中，创造出令人印象深刻的图像！

未经允许不得转载：大神网 » 如何使用Python快速去除图片背景并保存