From 64f025a818e18707aaa71e7bae9144827bb7f319 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 8 Jan 2019 20:59:17 +0100 Subject: thirds --- crop-thirds.py | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) (limited to 'crop-thirds.py') diff --git a/crop-thirds.py b/crop-thirds.py index 34f2336..c189b74 100644 --- a/crop-thirds.py +++ b/crop-thirds.py @@ -22,7 +22,7 @@ parser.add_argument('--vertical_offset', type=int, default=512) # parser.add_argument('--dst_height', type=int, default=512) parser.add_argument('--count', type=int, default=3) parser.add_argument('--aspect', type=float, default=3.0) -parser.add_argument('--folder_id', type=int, required=True) +# parser.add_argument('--folder_id', type=int, required=True) parser.add_argument('--overlap', type=float, default=0.5) parser.add_argument('--clobber', action='store_false') opt = parser.parse_args() @@ -35,14 +35,14 @@ dst_height = int(dst_width / opt.aspect) dst_size = (dst_width, dst_height,) offsets = [] +letters = [] -for i in range(count): - offsets.append(int(i / count * dst_width)) +for i in range(opt.count): + offsets.append(int(i / opt.count * dst_width)) + letters.append(chr(97 + i)) y0 = opt.vertical_offset - dst_height / 2 - -crops = [(x, y0, x + dst_width, d0 + dst_height,) for x in offsets] -letters = ['a', 'b', 'c'] +crops = [(x, y0, x + dst_width, y0 + dst_height,) for x in offsets] labels = [opt.label + '_' + letter for letter in letters] paths = [os.path.join(opt.out_dir, label) for label in labels] @@ -64,13 +64,11 @@ def build_thumbnail(i, fn): if (i % 100) == 0: print("{}...".format(i)) - canvas = Image.new('RGB', (int(src_width * 3/2), src_height,)) + canvas = Image.new('RGB', (int(src_width), src_height,)) image = Image.open(fn) - canvas.paste(image, (0, 0)) - canvas.paste(image, (src_width, 0)) for n, path in enumerate(paths): - canvas.crop(crops[n]).save(os.path.join(path, out_fn)) + image.crop(crops[n]).save(os.path.join(path, out_fn)) chunksize = 3 with Pool(processes=cpu_count()) as pool: -- cgit v1.2.3-70-g09d2