else:
grey = False
if args.recursive is not None:
rec = True
else:
rec = False
if grey:
IMAGE = cv2.imread(args.input, 0)
else:
IMAGE = cv2.imread(args.input, 1)
# Play with these arguments:
window_size = 4
gallery_im_siz = 32
gallery = Import.Importer(args.gallery, './output/', recursion=rec, im_size=gallery_im_siz, Gray=grey,
verbose=verbose)
ws = window_size
ncluster = min(len(gallery), (IMAGE.shape[0] * IMAGE.shape[1]) // ((ws + 1) ** 2))
C = Core.Core(IMAGE, window_size, n_cluster=ncluster)
# out, esum = C.build(gallery, target_pixel_size=args.scale // window_size)
out = C.build_cosine(gallery)
esum = 0
out = np.uint8(out)
print('Error = {}'.format(esum))
cv2.imwrite(args.output, out)
