import cv2, numpy as np
from sklearn.cluster import KMeans
def visualize_colors(cluster, centroids):
# Get the number of different clusters, create histogram, and normalize
labels = np.arange(0, len(np.unique(cluster.labels_)) + 1)
(hist, _) = np.histogram(cluster.labels_, bins = labels)
hist = hist.astype("float")
hist /= hist.sum()
# Create frequency rect and iterate through each cluster's color and percentage
rect = np.zeros((50, 300, 3), dtype=np.uint8)
colors = sorted([(percent, color) for (percent, color) in zip(hist, centroids)])
start = 0
for (percent, color) in colors:
print(color, "{:0.2f}%".format(percent * 100))
end = start + (percent * 300)
cv2.rectangle(rect, (int(start), 0), (int(end), 50), \
color.astype("uint8").tolist(), -1)
start = end
return rect
# Load image and convert to a list of pixels
image = cv2.imread('1.png')
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
reshape = image.reshape((image.shape[0] * image.shape[1], 3))
# Find and display most dominant colors
cluster = KMeans(n_clusters=5).fit(reshape)
visualize = visualize_colors(cluster, cluster.cluster_centers_)
visualize = cv2.cvtColor(visualize, cv2.COLOR_RGB2BGR)
cv2.imshow('visualize', visualize)
cv2.waitKey()
如何修正错误
发生此错误有两个潜在原因:
若要解决此问题,应确保文件名拼写正确(请执行区分大小写的检查,以防万一),并且图像文件位于当前工作目录中(此处有两个选项:您可以更改IDE中的当前工作目录或指定文件的完整路径)。
平均颜色与主色
然后要计算“平均颜色”,你必须决定你所指的是什么。在灰度图像中,它只是整个图像灰度的平均值,但对于颜色来说,没有所谓的“平均值”。实际上,颜色通常是通过三维矢量来表示的,而灰度则是标量。平均标量是可以的,但平均向量是没有意义的。
将图像分割成彩色分量并取其平均值是一种可能的方法。然而,这种方法可能会产生一种毫无意义的色彩。你可能真正想要的是主色而不是平均色。
实施
让我们慢慢地检查代码。我们首先导入必要的模块并读取图像:
然后,我们可以按照与@Ruan B提出的方法类似的方法计算每个色通道的平均值:
接下来,我们应用k-means clustering创建一个具有图像最具代表性颜色的调色板(在这个玩具示例中,
n_colors
被设置为5
)。最后,主色是在量化图像上出现频率最高的调色板颜色:
结果比较
为了说明这两种方法之间的差异,我使用了以下示例图像:
所获得的平均颜色值,即其成分是三个色通道平均值的颜色,以及通过k-平均值聚类计算出的主色是相当不同的:
让我们看看这些颜色看起来如何,以便更好地理解这两种方法之间的差异。下图左侧显示的是平均颜色。很明显,计算出的平均颜色不能正确地描述原始图像的颜色内容。事实上,在原始图像中没有一个像素有这种颜色。图的右侧显示了按重要性(出现频率)降序从上到下排序的五种最具代表性的颜色。这个调色板很明显地表明,主色调是红色,这与原始图像中最大的均匀颜色区域对应于红色乐高玩具的事实是一致的。
这是用于生成上图的代码:
TL;DR应答
总之,尽管平均颜色的计算——正如@Ruan B.的答案中所建议的那样——从数学角度来看在技术上是正确的,但得到的结果可能不能充分代表图像的颜色内容。一种更合理的方法是通过矢量量化(聚类)来确定主色。
另一种使用K-Means Clustering确定^{} 图像中主色的方法
输入图像
结果
对于
n_clusters=5
,这里是最主要的颜色和百分比分布每个颜色簇的可视化
与
n_clusters=10
相似我可以通过以下方法得到平均颜色:
结果:
Great Resource which I referenced
相关问题 更多 >
编程相关推荐