X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/3fa492032fb6f72fc8d153c0d11ad61155e8c1b9..e38a85551b486a0e1691eba67a9c8cfad528dbf1:/linef.py?ds=sidebyside diff --git a/linef.py b/linef.py index fbd1368..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") @@ -105,9 +109,9 @@ def find_lines(image, show_all, do_something, verbose): for line in [l for s in lines for l in s]: draw.line(line_from_angl_dist(line, image.size), fill=(120, 255, 120)) if show_all: - do_something(image_g, "the grid") + 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()