python - Python中的相似颜色检测

Question

给定 RGB 或 Hex 的颜色，我如何找到与其相似的颜色？通过相似，我的意思是它们应该可以通过小的值来区分。

Answer 1

RGB 颜色空间取决于设备，并且在感知上并不统一。如果您打算测量颜色相似性，则应首先将 RGB 值转换为与设备无关且在感知上更统一的颜色空间，例如CIELAB。然后，您可以通过适当的相似性度量来测量颜色差异，例如Lab Delta E。

演示

考虑这张图片：

让我们假设您的参考颜色是绿色阴影和洋红色阴影，其 RGB 值例如为[0, 160, 0]和[120, 0, 140]。下面的代码片段标识了那些相对于参考颜色的 Delta E 低于某个阈值（分别为15和）的图像像素。20

import numpy as np
from skimage import io
from skimage.color import rgb2lab, deltaE_cie76

rgb = io.imread('https://i.stack.imgur.com/npnrv.png')
lab = rgb2lab(rgb)

green = [0, 160, 0]
magenta = [120, 0, 140]

threshold_green = 15    
threshold_magenta = 20    

green_3d = np.uint8(np.asarray([[green]]))
magenta_3d = np.uint8(np.asarray([[magenta]]))

dE_green = deltaE_cie76(rgb2lab(green_3d), lab)
dE_magenta = deltaE_cie76(rgb2lab(magenta_3d), lab)

rgb[dE_green < threshold_green] = green_3d
rgb[dE_magenta < threshold_magenta] = magenta_3d
io.imshow(rgb)

Answer 2

根据 Tonechas 的回答，

要具有相似的颜色检测值（而不是可视化）：

import numpy as np
from skimage import io
from skimage.color import rgb2lab, deltaE_cie76

def get_pct_color(img_rgb, rgb_color, threshold=10):
    img_lab = rgb2lab(img_rgb)
    rgb_color_3d = np.uint8(np.asarray([[rgb_color]]))
    rgb_color_lab = rgb2lab(rgb_color_3d)
    delta = deltaE_cie76(rgb_color_lab, img_lab)
    x_positions, y_positions = np.where(delta < threshold)
    nb_pixel = img_rgb.shape[0] * img_rgb.shape[1]
    pct_color = len(x_positions) / nb_pixel
    return pct_color

举个例子：

img_rgb = io.imread('https://i.stack.imgur.com/npnrv.png')
green = [0, 160, 0]
get_pct_color(img_rgb, green, 10)
# 0.016131591796875
get_pct_color(img_rgb, green, 250)
# 1.0