clustering.py

# coding: utf-8
import sys
import numpy
from scipy.misc import imread, imsave
from scipy.cluster.vq import kmeans2

def main():
  if len(sys.argv) == 2:
    ifname = sys.argv[1]
  else:
    ifname = './shinchoku.jpg'

  k = 31

  src_image = imread(ifname)
  dots = src_image.reshape((src_image.shape[0]*src_image.shape[1], src_image.shape[2]))

  centroid, label = kmeans2(dots, k)
  centroid = centroid.astype(numpy.uint8)

  dst_image = centroid[label]
  dst_image = dst_image.reshape((src_image.shape[0], src_image.shape[1], src_image.shape[2]))

  imsave('dst_image.bmp', dst_image)

if __name__ == '__main__':
  main()