From 790d9cb736d8a5e30b6fafb0e4c00b7495199f03 Mon Sep 17 00:00:00 2001
From: Tomas Musil <tomik.musil@gmail.com>
Date: Tue, 10 Apr 2012 20:33:01 +0200
Subject: [PATCH] peak extraction

Much better results on the second test set.
---
 filters.py | 24 ++++++++++++++++++++++--
 imago.py   |  4 ++++
 2 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/filters.py b/filters.py
index df3b49f..222d8fe 100644
--- a/filters.py
+++ b/filters.py
@@ -10,8 +10,8 @@ def edge_detection(image):
     print "edge detection"
     for x in range(2, image.size[0] - 2):
         for y in range(2, image.size[1] - 2):
-            pix = (sum([sum(
-                [image_l[a, b] 
+            pix = (sum([sum([
+                image_l[a, b] 
                     for b in range(y - 2, y + 3)]) 
                     for a in range(x - 2, x + 3)])
                 - (25 * image_l[x, y]))
@@ -21,6 +21,26 @@ def edge_detection(image):
                 pix = 0 
             new_image_l[x, y] = pix
     return new_image
+	
+def peaks(image):
+    image_l = image.load()
+    new_image = Image.new('L', image.size)
+    new_image_l = new_image.load()
+    clear()
+    print "peak extraction"
+    for x in range(2, image.size[0] - 2):
+        for y in range(2, image.size[1] - 2):
+            pix = (sum([sum([
+                - image_l[a, b] 
+                    for b in range(y - 2, y + 3)]) 
+                    for a in range(x - 2, x + 3)])
+                + (16 * image_l[x, y]))
+            if pix > 255:
+                pix = 255
+            if pix < 0:
+                pix = 0 
+            new_image_l[x, y] = pix
+    return new_image
 
 def high_pass(image, height):
     image_l = image.load()
diff --git a/imago.py b/imago.py
index 3634363..45d165c 100755
--- a/imago.py
+++ b/imago.py
@@ -68,6 +68,10 @@ def main():
     im_hough = hough1.transform(im_h)
     if show_all:
         do_something(im_hough, "hough transform")
+		
+	im_hough = filters.peaks(im_hough)
+    if show_all:
+        do_something(im_hough, "peak extraction")
 
     im_h2 = filters.high_pass(im_hough, 120)
     if show_all:
-- 
2.4.2