k-means clustering

[imago.git] / intrsc.py

diff --git a/intrsc.py b/intrsc.py
index ba36b6d..97f9987 100644 (file)
--- a/intrsc.py
+++ b/intrsc.py
@@ -5,6 +5,8 @@ from operator import itemgetter
 
 import ImageDraw
 
 
 import ImageDraw
 

+import k_means
+

 def dst(line):
     """Return normalized line."""
     if line[0] < pi / 2:
 def dst(line):
     """Return normalized line."""
     if line[0] < pi / 2:


@@ -33,15 +35,59 @@ def board(image, lines, show_all, do_something):
                 draw.point((x , y), fill=(120, 255, 120))
         do_something(image_g, "intersections")
 
                 draw.point((x , y), fill=(120, 255, 120))
         do_something(image_g, "intersections")
 

-    board_r = []

     board_raw = []
     
     for line in intersections:
     board_raw = []
     
     for line in intersections:

-        board_r.append([stone_color(image, intersection) for intersection in
-                      line])

         board_raw.append([stone_color_raw(image, intersection) for intersection in
                       line])
         board_raw.append([stone_color_raw(image, intersection) for intersection in
                       line])

-    return board_r, board_raw
+    board_raw = sum(board_raw, [])
+
+    ### Show color distribution
+    luma = [(0.30 * s[0] + 0.59 * s[1] + 0.11 * s[2]) / 255.
+                 for s in board_raw]
+    saturation = [(max(s) - min(s)) / (255 - abs(max(s) + min(s) - 255))
+                 for s in board_raw]
+    if show_all:
+        import matplotlib.pyplot as pyplot
+        pyplot.scatter(luma, saturation, color=[(s[0]/255., s[1]/255., s[2]/255., 1.) 
+                                            for s in board_raw])
+        pyplot.show()
+
+    clusters = k_means.cluster(3, 2,zip(zip(luma, saturation), range(len(luma))),
+                               [[0., 0.], [0.5, 0.25], [1., 0.5]])
+   #clusters.sort(key=mean_luma)
+
+    if show_all:
+        pyplot.scatter([d[0][0] for d in clusters[0]], [d[0][1] for d in clusters[0]],
+                                                 color=(1,0,0,1))
+        pyplot.scatter([d[0][0] for d in clusters[1]], [d[0][1] for d in clusters[1]],
+                                                 color=(0,1,0,1))
+        pyplot.scatter([d[0][0] for d in clusters[2]], [d[0][1] for d in clusters[2]],
+                                                 color=(0,0,1,1))
+        pyplot.show()
+
+    clusters[0] = [(p[1], 'B') for p in clusters[0]]
+    clusters[1] = [(p[1], '.') for p in clusters[1]]
+    clusters[2] = [(p[1], 'W') for p in clusters[2]]
+
+    board_rl = sum(clusters, [])
+    board_rl.sort()
+    board_rg = (p[1] for p in board_rl)
+    
+    board_r = []
+
+    try:
+        for i in xrange(19):
+            board_r.append([])
+            for _ in xrange(19):
+                board_r[i].append(board_rg.next())
+    except StopIteration:
+        pass
+
+    return board_r
+
+def mean_luma(cluster):
+    return sum(c[0][0] for c in cluster) / float(len(cluster))

 
 def intersections_from_angl_dist(lines, size, get_all=True):
     """Take grid-lines and size of the image. Return intersections."""
 
 def intersections_from_angl_dist(lines, size, get_all=True):
     """Take grid-lines and size of the image. Return intersections."""


@@ -60,23 +106,6 @@ def intersections_from_angl_dist(lines, size, get_all=True):
         intersections.append(line)
     return intersections
    
         intersections.append(line)
     return intersections
    

-def stone_color(image, (x, y)):
-    """Given image and coordinates, return stone color."""
-    suma = 0.
-    for i in range(-2, 3):
-        for j in range(-2, 3):
-            try:
-                suma += sum(image.getpixel((x + i, y + j)))
-            except IndexError:
-                pass
-    suma /= 3 * 25
-    if suma < 55:
-        return 'B'
-    elif suma < 200: 
-        return '.'
-    else:
-        return 'W'
-   

 def stone_color_raw(image, (x, y)):
     """Given image and coordinates, return stone color."""
     suma = []
 def stone_color_raw(image, (x, y)):
     """Given image and coordinates, return stone color."""
     suma = []