7-1基于Kmeans的简单图像压缩

# -*- coding: utf-8 -*-

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

image = io.imread('./images/smile.jpg')
io.imshow(image)
io.show()

rows = image.shape[0]
cols = image.shape[1]

image = image.reshape(rows * cols, 3)
kmeans = KMeans(n_clusters=128, n_init= 10, max_iter=200)
kmeans.fit(image)

KMeans(algorithm='auto', copy_x=True, init='k-means++', max_iter=200,
    n_clusters=128, n_init=10, n_jobs=None, precompute_distances='auto',
    random_state=None, tol=0.0001, verbose=0)

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)

(128, 3)

io.imsave('./images/compressed_smile.jpg', labels)