Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/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)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement