Python中的主颜色检测

Question

我有大约 3000 张图像和 13 种不同的颜色（其中大多数图像的背景是白色）。如果图像的主颜色是这 13 种不同颜色之一，我希望将它们关联起来。

我见过类似的问题，例如 使用 python 进行图像颜色检测 要求平均颜色算法。我已经使用 Python 图像库和直方图复制了该代码，并使其正常工作 - 但我发现它对于确定主颜色不太可靠。

有什么想法吗？或者可以解决这个问题的图书馆？

提前致谢！

：编辑： 谢谢大家 - 你们几乎都说了同样的话，创建“桶”并用图像的每个最近像素增加桶计数。我似乎收到很多返回“白色”或“米色”的图像，这也是大多数图像的背景。有没有办法解决或忽略背景？

再次感谢。

Answer 1

您可以使用 getcolors 函数获取图像中所有颜色的列表。它返回以下形式的元组列表：

(N, COLOR)

其中 N 是颜色 COLOR 在图像中出现的次数。要获得最大出现的颜色，您可以将列表传递给 max 函数：

>>> from PIL import Image
>>> im = Image.open("test.jpg")
>>> max(im.getcolors(im.size[0]*im.size[1]))
(183, (255, 79, 79))

请注意，我将 im.size[0]*im.size[1] 传递给 getcolors 函数，因为这是最大 maxcolors 值（请参阅文档了解详细信息）。

Answer 2

就我个人而言，我会将颜色空间分为 8-16 个主要颜色，然后对于每个像素，我会将最接近的颜色桶增加 1。最后，像素数量最多的桶的颜色获胜。

基本上，考虑中位数而不是平均值。您只关心图像中的颜色，而平均颜色通常会给您一个全新的颜色。

Answer 3

由于您正在尝试匹配少量预先存在的颜色，因此您可以尝试不同的方法。针对所有颜色测试每张图像，看看哪一个最匹配。

至于进行匹配，我首先将每个图像调整为较小的尺寸，以减少您为每个图像所做的工作量；我们对图像颜色的感知不太依赖于细节的数量。对于较小图像的每个像素，找出 13 种颜色中最接近的一种。如果它在某个阈值内，则为该颜色增加一个计数器。最后，13个人中数最多的人获胜。