Python聚类色彩提取——Scipy-kmeans

一、聚类：物以类聚

数组可以进行聚类，并找到数组的聚类中心。使用的第三方库是scipy，需要pip install scipy，先安装该库。数组聚类代码：

import numpy as np
from scipy.cluster.vq import vq, kmeans, whiten
import matplotlib.pyplot as plt

fe = np.array([[1.9,2.0],
    [1.7,2.5],
    [1.6,3.1],
    [0.1,0.1],
    [0.8,0.3],
    [0.4,0.3],
    [0.22,0.1],
    [0.4, 0.3],
    [0.4,0.5],
    [1.8,1.9]])

book = np.array((fe[0], fe[1]))
print(type(book))
print("book:\n",book)

codebook, distortion = kmeans(fe,book)
print("codebook:",codebook)
print("distortion:",distortion)

plt.scatter(fe[:,0],fe[:,1],c="g")
plt.scatter(codebook[:,0],codebook[:,1],c="r")
plt.show()#红色的为聚类中心

运行结果：

 二、色彩聚类提取

图像颜色是由不同的RGB数值组成的，因此除了数字，图像色彩同样可以进行聚类。并利用之前学过的Pycharm flask框架，将结果加载到网页中显示。

思路是先生成图像缩略图，对缩略图聚类，得到每个点的RGB值，再利用算法将色彩聚类，并得到聚类中心。色彩聚类最常使用