from sklearn.datasets import load_sample_image import matplotlib.pyplot as plt from sklearn.cluster import KMeans import numpy as np flower=load_sample_image('flower.jpg')#原始图片 plt.imshow(flower) plt.show() image=flower[::3,::3]#降低原始图片的分辨率 plt.imshow(image) plt.show() #利用Kmeans对图片进行压缩 x=image.reshape(-1,3)#改变数组的形状 n_colors=64 model=KMeans(n_colors) labels=model.fit_predict(x) colors=model.cluster_centers_ new_image=colors[labels] new_image=new_image.reshape(image.shape) plt.imshow(new_image.astype(np.uint8)) plt.show()
