Image中对象的颜色检测

Question

我正在使用opencv和python。

我需要检测图像中物体的颜色，例如下图所示，衬衫颜色为红色。

在这个链接上，我发现了一些有用的东西，但它的皮肤检测图像。 http://lokeshdhakar.com/projects/color-thief/

我想我将不得不使用图像轮廓提取，然后进行颜色检测。

Answer 1

使用以下简单方法可以获得主色：

from sklearn.cluster import KMeans
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.patches as patches

img = cv2.imread('red_shirt.jpg')
height, width, dim = img.shape

编辑：仅拍摄图像的中心：

img = img[(height/4):(3*height/4), (width/4):(3*width/4), :]
height, width, dim = img.shape

img_vec = np.reshape(img, [height * width, dim] )

kmeans = KMeans(n_clusters=3)
kmeans.fit( img_vec )

编辑计算群集像素，按群组大小排序群集

unique_l, counts_l = np.unique(kmeans.labels_, return_counts=True)
sort_ix = np.argsort(counts_l)
sort_ix = sort_ix[::-1]

fig = plt.figure()
ax = fig.add_subplot(111)
x_from = 0.05

for cluster_center in kmeans.cluster_centers_[sort_ix]:
    ax.add_patch(patches.Rectangle( (x_from, 0.05), 0.29, 0.9, alpha=None,
                                    facecolor='#%02x%02x%02x' % (cluster_center[2], cluster_center[1], cluster_center[0] ) ) )
    x_from = x_from + 0.31

plt.show()

您可以使用this kind of preprocessing

删除BG和皮肤像素

Answer 2

1. 加载框架
2. 将BGR转换为HSV
3. 范围像素值

另请查看此链接Color detection in opencv