components and hough2 combined
[imago.git] / imago.py
index 45d165c..87b522e 100755 (executable)
--- a/imago.py
+++ b/imago.py
@@ -68,15 +68,19 @@ def main():
     im_hough = hough1.transform(im_h)
     if show_all:
         do_something(im_hough, "hough transform")
-               
-       im_hough = filters.peaks(im_hough)
+
+    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:
         do_something(im_h2, "second high pass filters")
 
+    im_h2 = filters.components(im_h2)
+    if show_all:
+        do_something(im_h2, "components centers")
+
     hough2 = Hough(im_h2.size)
     im_hough2 = hough2.transform(im_h2)
     if show_all:
@@ -93,7 +97,7 @@ def main():
     draw = ImageDraw.Draw(im_lines)
 
     for line in lines:
-        draw.line(line_from_angl_dist(line, im_h2.size), fill=255)
+        draw.line(line_from_angl_dist(line, im_h2.size), fill=255, width=5)
     if show_all:
         do_something(im_lines, "lines")