Kmeans进行图片压缩

Kmeans进行图片压缩

根据唐宇迪资料进行Kmeans的学习网上没有找到代码，所以把代码上传上来，供大家参考。侵权必删

1. 读取图片

from skimage import io
from sklearn.cluster import KMeans
import numpy as np

image = io.imread('a.jpg')
io.imshow(image)

2.压缩图片

rows = image.shape[0]
cols = image.shape[1]
image = image.reshape(image.shape[0]*image.shape[1],3) #一张图的像素点排成一列，3表示3个通道的值
kmeans = KMeans(n_clusters=128,n_init=10,max_iter=200)# 把256个值 分128个簇  迭代200次 
kmeans.fit(image)
clusters = np.asarray(kmeans.cluster_centers_,dtype=np.uint8) # 把聚类之后的中心给取出来
labels = np.asarray(kmeans.labels_,dtype=np.uint8)
labels = labels.reshape(rows,cols)#变成二维的了，所以是灰度图形式
io.imsave('compressed_test.jpg',labels)

3.展示压缩后的图片

image = io.imread('compressed_test.jpg')
io.imshow(image)