from tinydb import TinyDB from parallel import process import imageprocessing as ip import cv2, sys import numpy as np db = TinyDB(parse_args = False) db.arg_parser().add_argument('--image', required = True) db.arg_parser().add_argument('--filter', default = None) db.arg_parser().add_argument('--filterout', default = None) args = db.parse_args() d = 32 qi = ip.flatten_rgb_image(ip.read_rgb_image(args.image)) # ---------- 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') if args.filter == None: qi = np.int32(qi)
#!/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)
from tinydb import TinyDB from parallel import process import imageprocessing as ip import cv2, sys import numpy as np db = TinyDB(parse_args=False) db.arg_parser().add_argument('--image', required=True) db.arg_parser().add_argument('--filter', default=None) db.arg_parser().add_argument('--filterout', default=None) args = db.parse_args() d = 32 qi = ip.flatten_rgb_image(ip.read_rgb_image(args.image)) # ---------- 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')