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')
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)
