if image.size[0] > args.w:
image = image.resize((args.w, int((float(args.w)/image.size[0]) *
image.size[1])), Image.ANTIALIAS)
- do_something = im_debug.show
- if args.saving:
+
+ if not show_all:
+ def nothing(a, b):
+ pass
+ do_something = nothing
+ elif args.saving:
do_something = Imsave("saved/" + args.files[0][:-4] + "_" +
str(image.size[0]) + "/").save
+ else:
+ do_something = im_debug.show
if args.manual_mode:
try:
pickle.dump((lines, l1, l2, bounds, hough), d_file)
d_file.close()
else:
- lines, l1, l2, bounds, hough = linef.find_lines(image, show_all, do_something, verbose)
+ lines, l1, l2, bounds, hough = linef.find_lines(image, do_something, verbose)
grid, lines = gridf.find(lines, image.size, l1, l2, bounds, hough,
show_all, do_something)
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 = filters.high_pass(im_hough, 96)
show_image(im_h2, "second high pass filters")
return im_h2
-def find_lines(image, show_all, do_something, verbose):
+def find_lines(image, show_image, verbose):
"""Find lines in the *image*."""
# TODO refactor into smaller functions
- def nothing(a, b):
- pass
-
- if show_all:
- show_image = do_something
- else:
- show_image = nothing
-
if verbose:
print >> sys.stderr, "preprocessing"