vector quantization

from scipy import misc
from sklearn import cluster
import numpy

face = misc.face(gray=True)
x = face.reshape((-1,1))
kmeans = cluster.KMeans(n_clusters=5)
kmeans.fit(x)
labels = kmeans.labels_
values = kmeans.cluster_centers_.squeeze()
compress_face = numpy.choose(labels, values)
compress_face.shape = face.shape
# print(compress_face)