X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/0548d977be411cbd79451c53d4507603df87d128..666ac3553431948203df831afe240beb3fcaec32:/src/linef.py?ds=sidebyside

diff --git a/src/linef.py b/src/linef.py
index 8ebb702..8de87a6 100644
--- a/src/linef.py
+++ b/src/linef.py
@@ -5,7 +5,7 @@ import sys
 from math import sin, cos, pi
 
 try:
-    import Image, ImageDraw
+    from PIL import Image, ImageDraw
 except ImportError, msg:
     print >> sys.stderr, msg
     sys.exit(1)
@@ -56,6 +56,8 @@ def run_ransac(image):
         for x in xrange(0, width):
             if image_l[x, y] > 128:
                 data.append((x, y))
+                if y < 30:
+                    data.append((width - x, y + height))
 
     dist = 3 
     [(line, points), (line2, points2)] = ransac.ransac_multi(2, data, dist, 250)