X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/f6002686fef1bd826f3a1777998416b93d909d56..db4fb6da5dc88a3944352cff38cb92f68bead88c:/linef.py diff --git a/linef.py b/linef.py index 34c4d07..37690cc 100755 --- a/linef.py +++ b/linef.py @@ -83,6 +83,7 @@ def find_lines(image, show_all, do_something, verbose): lines = [] im_c = im_h2.convert('RGB').convert('RGB', (1, 0.5, 0.5, 0)) draw_c = ImageDraw.Draw(im_c) + bounds = [] for line_l in lines_m: im_line = Image.new('L', im_h2.size) @@ -96,6 +97,9 @@ def find_lines(image, show_all, do_something, verbose): for point in line_points: draw_c.point(point, fill=(120, 255, 120)) lines.append(hough1.lines_from_list(line_points)) + line_points = list(line_points) + line_points.sort() + bounds += [line_points[0], line_points[-1]] if show_all: do_something(im_c, "hough x lines") @@ -107,7 +111,7 @@ def find_lines(image, show_all, do_something, verbose): if show_all: do_something(image_g, "lines") - return lines + return lines, lines_m[0][0], lines_m[1][0], bounds, hough1 def combine(image1, image2): im_l1 = image1.load()