Python PIL比较颜色

Question

我有一个像这样嘈杂的背景图像（炸毁，每个方块都是一个像素）。我正在尝试将黑色背景标准化，以便我可以完全替换颜色。

这就是我的想法（伪代码）：

for pixel in image:
    if is_similar(pixel, (0, 0, 0), threshold):
        pixel = (0, 0, 0)

哪种功能可以让我比较两个颜色值以匹配某个阈值？

Answer 1

我最终使用了来自this answer的感知亮度公式。它运作得很好。

THRESHOLD = 18

def luminance(pixel):
    return (0.299 * pixel[0] + 0.587 * pixel[1] + 0.114 * pixel[2])


def is_similar(pixel_a, pixel_b, threshold):
    return abs(luminance(pixel_a) - luminance(pixel_b)) < threshold


width, height = img.size
pixels = img.load()

for x in range(width):
    for y in range(height):
        if is_similar(pixels[x, y], (0, 0, 0), THRESHOLD):
            pixels[x, y] = (0, 0, 0)