def do_filter(arr, filt): if filt == None: return arr elif filt == 'raw,sobel': i = ip.unflatten_rgb_image(arr, d, d) i = ip.sobel_scipy(i) i = ip.gray_as_rgb(i) return ip.flatten_rgb_image(i) raise Exception('unknown filter')
#!/usr/bin/env python import argparse from imageprocessing import read_rgb_image, write_rgb_image, gray_as_rgb, sobel_scipy, rgb_as_gray, sobel p = argparse.ArgumentParser() p.add_argument('-i', required = True) p.add_argument('-o', required = True) p.add_argument('-f', required = True) args = p.parse_args() i = read_rgb_image(args.i) o = None if args.f == 'sobel': o = gray_as_rgb(sobel(i)) elif args.f == 'gray': o = gray_as_rgb(rgb_as_gray(i)) else: raise Exception('unknown filter') write_rgb_image(args.o, o)