Python PIL - 寻找最接近的颜色(舍入颜色)

Python PIL - Finding Nearest Color (Rounding Colors)

有一款名为 Roblox 的游戏,玩家通常使用乐高积木制作游戏。

在 Roblox 中,对于 Brick 颜色,您可以使用典型的 RGB 值,但这需要一个额外的元素,在文件大小方面效率不是很高。与使用 RGB 值不同,BrickColor 在文件大小方面更为经济。它使用一个整数来解释为某种颜色。 Here is what I mean:

这是我的代码片段:

import Image
f = raw_input('Image:\n')
im = Image.open(f)
rgb_im = im.convert('RGB')
r, g, b = rgb_im.getpixel((x, y))

在我的程序中,我需要找到最接近 RGB 值的颜色代码。

如何实现?

在 table 中创建一个颜色列表(我称之为颜色)。 按到您要查询的 r、g、b 点的距离对列表进行排序 列表中的第一个元素是最接近的颜色

def distance(c1, c2):
    (r1,g1,b1) = c1
    (r2,g2,b2) = c2
    return math.sqrt((r1 - r2)**2 + (g1 - g2) ** 2 + (b1 - b2) **2)

colors = list(rgb_code_dictionary.keys())
closest_colors = sorted(colors, key=lambda color: distance(color, point))
closest_color = closest_colors[0]
code = rgb_code_dictionary[closest_color]

扩展 mattsap 的回答:

我们不需要对所有颜色进行排序,因为我们只是在寻找最接近的。即我们可以避免计算上昂贵的 sort 并使用 min 代替。

我们也不需要计算颜色之间的绝对距离,因为我们只对相对距离感兴趣。即我们也可以避免毕达哥拉斯的 "square root" 部分。

这给出:

colours = ( (255, 255, 255, "white"),
            (255, 0, 0, "red"),
            (128, 0, 0, "dark red"),
            (0, 255, 0, "green") )


def nearest_colour( subjects, query ):
    return min( subjects, key = lambda subject: sum( (s - q) ** 2 for s, q in zip( subject, query ) ) )


print( nearest_colour( colours, (64, 0, 0) ) ) # dark red
print( nearest_colour( colours, (0, 192, 0) ) ) # green
print( nearest_colour( colours, (255, 255, 64) ) ) # white

当然,一旦您考虑了不同的颜色空间以及每个颜色分量对人眼感知的贡献,就会有一个完整的兔子洞,as per this question,但这对于大多数情况下。