用聚类分析将图片中所有的颜色值做聚类。
得到的聚类中心的颜色替换原来位置的颜色,形成新的图片。
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