图片颜色聚类分析--KMeans

用聚类分析将图片中所有的颜色值做聚类。
得到的聚类中心的颜色替换原来位置的颜色，形成新的图片。

from skimage import io
import matplotlib.image as img
from sklearn.cluster import KMeans
import numpy as np
from matplotlib import pyplot as plt

image = io.imread("E:/picture.jpg")
X = image.reshape(-1,3)  #-1表示自动通过总量和列数3补齐行数，3为颜色三元素

# 颜色聚类划分
mod = KMeans(n_clusters = 16)  #聚类个数
labels = mod.fit_predict(X