python - 相同图像的不同背景但大小不同

Question

我有一些产品图片。这些图像的背景是白色的。现在，在我的网站中，我想在不同的地方使用不同尺寸的相同图像。问题是，由于图像背景的颜色是白色。但是，我需要作为缩略图的背景颜色是绿色。同样，作为主要产品图像，我希望背景为浅蓝色。例如，这是一个图像。 链接。你可以看到这张图片的背景是白色的。我想使用相同的图像作为缩略图。我编写了一个代码来生成图像的缩略图。但我想要背景作为其他颜色。这可以通过python中的图像处理来完成吗？谢谢你。

编辑：我不能评论任何答案，不明白为什么。请回复。

Answer 1

如果您可以容忍生成的图像中出现一些丑陋的效果，这很容易做到。如果这张具有不同背景颜色的图像将显示为原始图像的缩小版本，那么这些效果可能不会很明显，一切都很好。

所以这里是简单的方法：

• 假设它是背景像素（在您的示例中为白色），则从 (0, 0) 处的像素进行填充填充，并在执行填充填充时接受细微的差异。背景像素被透明点替换。
• 上面的步骤给出了一个掩码，例如，您可以执行腐蚀和高斯滤波。
• 使用上面创建的蒙版粘贴“洪水填充”图像。

以下是您可以从这种方法中获得的期望。输入图像，然后两次转换为不同的背景颜色。

import sys
import cv2
import numpy
from PIL import Image

def floodfill(im, grayimg, seed, color, tolerance=15):
    width, height = grayimg.size
    grayim = grayimg.load()
    start_color = grayim[seed]

    mask_img = Image.new('L', grayimg.size, 255)
    mask = mask_img.load()
    count = 0
    work = [seed]
    while work:
        x, y = work.pop()
        im[x, y] = color
        for dx, dy in ((-1,0), (1,0), (0,-1), (0,1)):
            nx, ny = x + dx, y + dy
            if nx < 0 or ny < 0 or nx > width - 1 or ny > height - 1:
                continue
            if mask[nx, ny] and abs(grayim[nx, ny] - start_color) <= tolerance:
                mask[nx, ny] = 0
                work.append((nx, ny))
    return mask_img

img = Image.open(sys.argv[1]).convert('RGBA')
width, height = img.size
img_p = Image.new('RGBA', (width + 20, height + 20), img.getpixel((0, 0)))
img_p.paste(img, (3, 3))
img = img_p
img_g = img.convert('L')
width, height = img.size

im = img.load()
mask = floodfill(im, img_g, (0, 0), (0, 0, 0, 0), 20)

mask = numpy.array(mask)
se = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (7, 7))
mask = cv2.erode(mask, se)
mask = cv2.GaussianBlur(mask, (9, 9), 3)
mask = Image.fromarray(mask)

result_bgcolor = (0, 0, 0, 255) # Change to match the color you wish.
result = Image.new('RGBA', (width, height), result_bgcolor)
result.paste(img_p, (0, 0), mask)

result.save(sys.argv[2])