quantify-kmeans.py

#!/usr/bin/env python2

from sys import argv
from PIL import Image
from scipy.cluster.vq import kmeans2
import numpy

_, path_in, k, path_out = argv
img = Image.open(path_in).convert('RGB')
centroid, label = kmeans2(numpy.array(img.getdata()), int(k))
centroid = numpy.around(centroid).astype('int')
img.putdata([tuple(centroid[l]) for l in label])
img.save(path_out)