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)