from skimage import io
from sklearn.cluster import KMeans
import numpy as np
image = io.imread('企鹅.png')
io.imshow(image)
io.show()
rows = image.shape[0]
cols = image.shape[1]
image = image.reshape(image.shape[0]*image.shape[1],3)
Kmeans = KMeans(n_clusters = 128,n_init=10, max_iter=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)
print(clusters.shape)
np.save('codebook_test.npy',clusters)
io.imsave('compressed_test.png',labels)
python 使用Kmeans进行图像压缩
最新推荐文章于 2022-07-29 18:09:10 发布