基于Python查找图像中很常见的颜色

 如果我们能够得知道一幅图像中最多的颜色是什么的话，可以帮助我们解决很多实际问题。例如在农业领域中想确定水果的成熟度，我们可以通过检查水果的颜色是否落在特定范围内，来判断它们是否已经成熟。

景德镇ssl适用于网站、小程序/APP、API接口等需要进行数据传输应用场景，ssl证书未来市场广阔！成为成都创新互联公司的ssl证书销售渠道，可以享受市场价格4-6折优惠！如果有意向欢迎电话联系或者加微信：028-86922220（备注：SSL证书合作）期待与您的合作！

接下来我们将使用Python和一些常用库(例如Numpy，Matplotlib和OpenCV)来解决这个问题。

01. 准备工作

第一步：添加程序包

我们将在此处加载基本软件包。另外，由于我们要使用Jupyter进行编程，因此小伙伴们不要忘记添加%matplotlib inline命令。

第二步：加载并显示示例图像

我们将并排显示两个图像，因此我们需要做一个辅助函数。接下来我们将加载一些在本教程中将要使用的示例图像，并使用上述功能对其进行显示。

02. 常用方法

方法一：平均值

第一种方法是最简单(但无效)的方法-只需找到平均像素值即可。使用numpy的average功能，我们可以轻松获得行和宽度上的平均像素值-axis=(0,1)

 
 
 
 
    
  
  
  
  
  1. img_temp = img.copy()
    2. 
  
  
  
  
  2. img_temp[:,:,0], img_temp[:,:,1], img_temp[:,:,2] = np.average(img, axis=(0,1))
    3. 
  
  
  
  
  3. 
  
  
  
  
  4. img_temp_2 = img_2.copy()
    5. 
  
  
  
  
  5. img_temp_2[:,:,0], img_temp_2[:,:,1], img_temp_2[:,:,2] = np.average(img_2, axis=(0,1))
    6. 
  
  
  
  
  6. 
  
  
  
  
  7. show_img_compar(img, img_temp)
    8. 
  
  
  
  
  8. show_img_compar(img_2, img_temp_2)
    9.

从上面图像中可以看出，平均方法可能会产生错误结果，它给出的最常见的颜色可能并不是我们想要的颜色，这是因为平均值考虑了所有像素值。当我们具有高对比度的图像(一张图像中同时包含“浅色”和“深色”)时这个问题会很严重。在第二张图片中，这一点更加清晰。它为我们提供了一种新的颜色，该颜色在图像中根本看不到。

方法二：最高像素频率

第二种方法将比第一种更加准确。我们的工作就是计算每个像素值出现的次数。numpy给我们提供了一个函数可以完成这个任务。但是首先，我们必须调整图像数据结构的形状，以仅提供3个值的列表(每个R，G和B通道强度一个)。

我们可以使用numpy的reshape函数来获取像素值列表。

现在我们已经有了正确结构的数据，可以开始计算像素值的频率了，使用numpy中的unique函数即可。

 
 
 
 
    
  
  
  
  
  1. img_temp = img.copy()
    2. 
  
  
  
  
  2. unique, counts = np.unique(img_temp.reshape(-1, 3), axis=0, return_counts=True)
    3. 
  
  
  
  
  3. img_temp[:,:,0], img_temp[:,:,1], img_temp[:,:,2] = unique[np.argmax(counts)]
    4. 
  
  
  
  
  4. 
  
  
  
  
  5. img_temp_2 = img_2.copy()
    6. 
  
  
  
  
  6. unique, counts = np.unique(img_temp_2.reshape(-1, 3), axis=0, return_counts=True)
    7. 
  
  
  
  
  7. img_temp_2[:,:,0], img_temp_2[:,:,1], img_temp_2[:,:,2] = unique[np.argmax(counts)]
    8. 
  
  
  
  
  8. 
  
  
  
  
  9. show_img_compar(img, img_temp)
    10. 
  
  
  
  
  10. show_img_compar(img_2, img_temp_2)
    11.

比第一个更有意义吗?最常见的颜色是黑色区域。但是如果我们不仅采用一种最常见的颜色，还要采用更多的颜色怎么办?使用相同的概念，我们可以采用N种最常见的颜色。换句话说，我们要采用最常见的不同颜色群集该怎么办。

方法三：使用K均值聚类

我们可以使用著名的K均值聚类将颜色组聚类在一起。

 
 
 
 
    
  
  
  
  
  1. def palette(clusters):
    2. 
  
  
  
  
  2.     width=300
    3. 
  
  
  
  
  3.     palette = np.zeros((50, width, 3), np.uint8)
    4. 
  
  
  
  
  4.     steps = width/clusters.cluster_centers_.shape[0]
    5. 
  
  
  
  
  5.     for idx, centers in enumerate(clusters.cluster_centers_): 
    6. 
  
  
  
  
  6.         palette[:, int(idx*steps):(int((idx+1)*steps)), :] = centers
    7. 
  
  
  
  
  7.     return palette
    8. 
  
  
  
  
  8. 
  
  
  
  
  9. 
  
  
  
  
  10. clt_1 = clt.fit(img.reshape(-1, 3))
    11. 
  
  
  
  
  11. show_img_compar(img, palette(clt_1))
    12. 
  
  
  
  
  12. 
  
  
  
  
  13. clt_2 = clt.fit(img_2.reshape(-1, 3))
    14. 
  
  
  
  
  14. show_img_compar(img_2, palette(clt_2))
    15.

容易吧!现在，我们需要的是一个显示上面的颜色簇并立即显示的功能。我们只需要创建一个高度为50，宽度为300像素的图像来显示颜色组/调色板。对于每个颜色簇，我们将其分配给我们的调色板。

是不是很漂亮?就图像中最常见的颜色而言，K均值聚类给出了出色的结果。在第二张图像中，我们可以看到调色板中有太多的棕色阴影。这很可能是因为我们选择了太多的群集。让我们看看是否可以通过选择较小的k值来对其进行修复。

 
 
 
 
    
  
  
  
  
  1. def palette(clusters):
    2. 
  
  
  
  
  2.     width=300
    3. 
  
  
  
  
  3.     palette = np.zeros((50, width, 3), np.uint8)
    4. 
  
  
  
  
  4.     steps = width/clusters.cluster_centers_.shape[0]
    5. 
  
  
  
  
  5.     for idx, centers in enumerate(clusters.cluster_centers_): 
    6. 
  
  
  
  
  6.         palette[:, int(idx*steps):(int((idx+1)*steps)), :] = centers
    7. 
  
  
  
  
  7.     return palette
    8. 
  
  
  
  
  8. 
  
  
  
  
  9. clt_3 = KMeans(n_clusters=3)
    10. 
  
  
  
  
  10. clt_3.fit(img_2.reshape(-1, 3))
    11. 
  
  
  
  
  11. show_img_compar(img_2, palette(clt_3))
    12.

由于我们使用K均值聚类，因此我们仍然必须自己确定适当数量的聚类。三个集群似乎是一个不错的选择。但是我们仍然可以改善这些结果，并且仍然可以解决集群问题。我们还如何显示群集在整个图像中所占的比例?

方法四：K均值+比例显示

我们需要做的就是修改我们的palette功能。代替使用固定步骤，我们将每个群集的宽度更改为与该群集中的像素数成比例。

 
 
 
 
    
  
  
  
  
  1. from collections import Counter
    2. 
  
  
  
  
  2. 
  
  
  
  
  3. def palette_perc(k_cluster):
    4. 
  
  
  
  
  4.     width = 300
    5. 
  
  
  
  
  5.     palette = np.zeros((50, width, 3), np.uint8)
    6. 
  
  
  
  
  6. 
  
  
  
  
  7.     n_pixels = len(k_cluster.labels_)
    8. 
  
  
  
  
  8.     counter = Counter(k_cluster.labels_) # count how many pixels per cluster
    9. 
  
  
  
  
  9.     perc = {}
    10. 
  
  
  
  
  10.     for i in counter:
    11. 
  
  
  
  
  11.         perc[i] = np.round(counter[i]/n_pixels, 2)
    12. 
  
  
  
  
  12.     perc = dict(sorted(perc.items()))
    13. 
  
  
  
  
  13. 
  
  
  
  
  14.     #for logging purposes
    15. 
  
  
  
  
  15.     print(perc)
    16. 
  
  
  
  
  16.     print(k_cluster.cluster_centers_)
    17. 
  
  
  
  
  17. 
  
  
  
  
  18.     step = 0
    19. 
  
  
  
  
  19. 
  
  
  
  
  20.     for idx, centers in enumerate(k_cluster.cluster_centers_): 
    21. 
  
  
  
  
  21.         palette[:, step:int(step + perc[idx]*width+1), :] = centers
    22. 
  
  
  
  
  22.         step += int(perc[idx]*width+1)
    23. 
  
  
  
  
  23. 
  
  
  
  
  24.     return palette
    25. 
  
  
  
  
  25. 
  
  
  
  
  26. clt_1 = clt.fit(img.reshape(-1, 3))
    27. 
  
  
  
  
  27. show_img_compar(img, palette_perc(clt_1))
    28. 
  
  
  
  
  28. 
  
  
  
  
  29. clt_2 = clt.fit(img_2.reshape(-1, 3))
    30. 
  
  
  
  
  30. show_img_compar(img_2, palette_perc(clt_2))
    31.

它不仅为我们提供了图像中最常见的颜色。这也给了我们每个像素出现的比例。

03. 结论

我们介绍了几种使用Python以及最老牌的库来获取图像中最常见颜色的技术。另外，我们还看到了这些技术的优缺点。到目前为止，使用k> 1的K均值找到最常见的颜色是找到图像中最频繁的颜色的最佳解决方案之一。

网站栏目：基于Python查找图像中很常见的颜色
当前网址：http://www.hantingmc.com/qtweb/news34/23834.html

网站建设、网络推广公司-创新互联，是专注品牌与效果的网站制作，网络营销seo公司；服务项目有等

广告

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源： 创新互联

猜你还喜欢下面的内容

网站收录知识

同城分类信息

四川成都做网站　　 　简阳网站建设　　 　网络推广公司　　 　泸州网站建设　　 　重庆外贸网站建设　　 　成都服务器托管　　 　绵阳服务器托管　　 　卡多尼人工智　　 　云服务器　　 　成都400电话　　 　成都网络营销　　 　郫县门窗定制　　 　做网站公司　　 　成都木托盘　　 　企业网站建设公司　　 　成都棕树机房　　 　H5网站制作　　 　大足网站建设　　 　德信企业管理　　 　做网站