コード例 #1
0
ファイル: knn.py プロジェクト: daniel-e/tinyimages
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)
コード例 #2
0
#!/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)
コード例 #3
0
ファイル: knn.py プロジェクト: daniel-e/tinyimages
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')