fix split lines in Hough space
authorTomas Musil <tomik.musil@gmail.com>
Wed, 23 Jul 2014 16:17:46 +0000 (18:17 +0200)
committerTomas Musil <tomik.musil@gmail.com>
Wed, 23 Jul 2014 16:17:46 +0000 (18:17 +0200)
src/linef.py

index aa9921b..8de87a6 100644 (file)
@@ -56,6 +56,8 @@ def run_ransac(image):
         for x in xrange(0, width):
             if image_l[x, y] > 128:
                 data.append((x, y))
         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)
 
     dist = 3 
     [(line, points), (line2, points2)] = ransac.ransac_multi(2, data, dist, 250)