X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/3790db9ab7a4871faf0b445c8283c7f0b7b30a22..16a9526d24fbdad7bbc544aeae9fd7ba067b3d91:/imago_pack/linef.py?ds=sidebyside diff --git a/imago_pack/linef.py b/imago_pack/linef.py index eda82ce..ab7c95d 100644 --- a/imago_pack/linef.py +++ b/imago_pack/linef.py @@ -1,4 +1,4 @@ -"""Go image recognition lines-finding module""" +"""Lines finding module.""" from functools import partial import sys @@ -12,118 +12,97 @@ except ImportError, msg: import filters from hough import Hough +import ransac -def find_lines(image, show_all, do_something, verbose): - - if verbose: - print >> sys.stderr, "preprocessing" - - if show_all: - do_something(image, "original image") - +def prepare(image, show_image, logger): + # TODO comment im_l = image.convert('L') - if show_all: - do_something(im_l, "ITU-R 601-2 luma transform") - - if verbose: - print >> sys.stderr, "edge detection" + show_image(im_l, "ITU-R 601-2 luma transform") + logger("edge detection") im_edges = filters.edge_detection(im_l) - if show_all: - do_something(im_edges, "edge detection") + show_image(im_edges, "edge detection") im_h = filters.high_pass(im_edges, 100) - if show_all: - do_something(im_h, "high pass filters") - - if verbose: - print >> sys.stderr, "hough transform" + show_image(im_h, "high pass filters") - im_hough = Hough.Transform(im_h) - if show_all: - do_something(im_hough.image, "hough transform") + return im_h + +def transform(image, hough, show_image): + # TODO comment + im_hough = hough.transform(image) + show_image(im_hough, "hough transform") # im_hough.image = filters.peaks(im_hough.image) - # if show_all: - # do_something(im_hough.image, "peak extraction") + # show_image(im_hough.image, "peak extraction") - im_h2 = im_hough.apply_filter(partial(filters.high_pass, height=96)) - if show_all: - do_something(im_h2.image, "second high pass filters") - - im_h2 = im_h2.apply_filter(filters.components2) - if show_all: - do_something(im_h2.image, "components centers") - - if verbose: - print >> sys.stderr, "second hough transform" - - # im_hough might be used instead im_h2, but at the moment it brings a lot of - # noise to the second transform, which later confuses the center-finding - # mechanism (which is not very robust yet) - im_hough2 = Hough.Transform(im_h2.image) - if show_all: - do_something(im_hough2.image, "second hough transform") - - im_h3 = im_hough2.apply_filter(partial(filters.high_pass, height=120)) - if show_all: - do_something(im_h3.image, "third high pass filter") - - im_h3 = im_h3.apply_filter(filters.components) - if show_all: - do_something(im_h3.image, "half centers") - - if verbose: - print >> sys.stderr, "finding the grid" - - lines_m = im_h3.all_lines_h() - lines = [] - im_c = im_h2.image.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) - draw = ImageDraw.Draw(im_line) - line_points = set() - for line in line_l: - draw.line(line_from_angl_dist(line, im_h2.size), fill=255, width=7) - draw_c.line(line_from_angl_dist(line, im_c.size), - fill=(70, 70, 70), width=7) - for p in combine(im_h2.image, im_line): - line_points.add(p) - for point in line_points: - draw_c.point(point, fill=(120, 255, 120)) - lines.append(im_hough.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") + im_h2 = filters.high_pass(im_hough, 96) + show_image(im_h2, "second high pass filters") + + im_h2 = filters.components(im_h2, 2) + show_image(im_h2, "components centers") + + return im_h2 + +def run_ransac(image): + # TODO comment + # TODO vizualize this + image_l = image.load() + width, height = image.size + + data = [] + + for y in xrange(0, height): + for x in xrange(0, width): + if image_l[x, y] > 128: + data.append((x, y)) + + dist = 3 + (line, points), (line2, points2) = ransac.ransac_duo(data, dist, 75, 15) + line_to_points = lambda (a, b, c), x: (x, (a*x + c) / (- b)) + # TODO width should not be here vvv + # TODO refactor gridf to use standard equations instead of points + line = [line_to_points(line, 0), line_to_points(line, width - 1)] + line2 = [line_to_points(line2, 0), line_to_points(line2, width - 1)] + return [sorted(points), sorted(points2)], line, line2 + +def find_lines(image, show_image, logger): + """Find lines in the *image*.""" + + logger("preprocessing") + show_image(image, "original image") + + im_h = prepare(image, show_image, logger) + + hough = Hough.default(im_h) + logger("hough transform") + + im_h2 = transform(im_h, hough, show_image) + + logger("finding the lines") + + r_lines, l1, l2 = run_ransac(im_h2) + + lines = map(hough.lines_from_list, r_lines) + + # TODO refactor gridf to get rid of this: + bounds = sum(map(lambda l: [l[0], l[-1]], r_lines), []) + # sum(list, []) = flatten list + + # TODO do this only if show_all is true: image_g = image.copy() draw = ImageDraw.Draw(image_g) 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, "lines") - - return lines, lines_m[0][0], lines_m[1][0], bounds, im_hough - -def combine(image1, image2): - im_l1 = image1.load() - im_l2 = image2.load() + show_image(image_g, "lines") - on_both = [] - for x in xrange(image1.size[0]): - for y in xrange(image1.size[1]): - if im_l1[x, y] and im_l2[x, y]: - on_both.append((x, y)) - return on_both + return lines, l1, l2, bounds, hough # TODO def line_from_angl_dist((angle, distance), size): + """Take *angle* and *distance* (from the center of the image) of a line and + size of the image. Return the line represented by two points.""" if pi / 4 < angle < 3 * pi / 4: y1 = - size[1] / 2 x1 = int(round((y1 * cos(angle) + distance) / sin(angle))) + size[0] / 2