如何使用 PIL 在单个白色背景上检测和裁剪多个图像？

Posted 2023-02-25

【中文标题】如何使用 PIL 在单个白色背景上检测和裁剪多个图像？

【英文标题】：How do you detect and crop multiple images on a single white background using PIL?

【发布时间】：2018-07-05 22:21:57

【问题描述】：

我刚开始使用 PIL，我需要帮助来使用 PIL 检测和裁剪单个白色背景上的多个图像。它们可以是不同的大小并且位于不同的位置。现在，我只能裁剪出一张图像。 请帮忙，谢谢！

def trim(im):
    bg = Image.new(im.mode, im.size, im.getpixel((0,0)))
    diff = ImageChops.difference(im, bg)
    diff = ImageChops.add(diff, diff, 2.0, -100)
    bbox = diff.getbbox()
    if bbox:
        return im.crop(bbox)
    else:
        print("No image detected")

image1 = Image.open('multiple.jpg')
image2 = test(image1)

【问题讨论】：

请编辑您的问题，以便您的代码是一个最小完整的可验证示例，而不是一个片段。并对其进行测试以确保它显示您的问题。 ***.com/mcve 提示：没有调用您的 trim() 函数，至少是一个问题。

【参考方案1】：

这可以提高效率，但这会使答案复杂化。

from PIL import Image, ImageChops

def crop(im, white):
    bg = Image.new(im.mode, im.size, white)
    diff = ImageChops.difference(im, bg)
    bbox = diff.getbbox()
    if bbox:
        return im.crop(bbox)

def split(im, white):
    # Is there a horizontal white line?
    whiteLine = Image.new(im.mode, (im.width, 1), white)
    for y in range(im.height):
        line = im.crop((0, y, im.width, y+1))
        if line.tobytes() == whiteLine.tobytes():
            # There is a white line
            # So we can split the image into two
            # and for efficiency, crop it
            ims = [
                crop(im.crop((0, 0, im.width, y)), white),
                crop(im.crop((0, y+1, im.width, im.height)), white)
            ]
            # Now, because there may be white lines within the two subimages
            # Call split again, making this recursive
            return [sub_im for im in ims for sub_im in split(im, white)]

    # Is there a vertical white line?
    whiteLine = Image.new(im.mode, (1, im.height), white)
    for x in range(im.width):
        line = im.crop((x, 0, x+1, im.height))
        if line.tobytes() == whiteLine.tobytes():
            ims = [
                crop(im.crop((0, 0, x, im.height)), white),
                crop(im.crop((x+1, 0, im.width, im.height)), white)
            ]
            return [sub_im for im in ims for sub_im in split(im, white)]

    # There are no horizontal or vertical white lines
    return [im]

def trim(im):
    # You have defined the pixel at (0, 0) as white in your code
    white = im.getpixel((0,0))

    # Initial crop
    im = crop(im, white)
    if im:
        return split(im, white)
    else:
        print("No image detected")

image1 = Image.open('multiple.jpg')
trim(image1)