diff options
Diffstat (limited to 'crop-thirds.py')
| -rw-r--r-- | crop-thirds.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/crop-thirds.py b/crop-thirds.py index e2809a2..900f6f4 100644 --- a/crop-thirds.py +++ b/crop-thirds.py @@ -20,6 +20,7 @@ parser.add_argument('--vertical_offset', type=int, default=256) # parser.add_argument('--dst_height', type=int, default=512) parser.add_argument('--count', type=int, default=3) parser.add_argument('--max', type=int, default=99997) +parser.add_argument('--crop_aspect' type=float, default=1.5) parser.add_argument('--aspect', type=float, default=3.0) # parser.add_argument('--folder_id', type=int, required=True) parser.add_argument('--margin', type=int, default=16) @@ -31,6 +32,11 @@ src_height = 512 dst_width = int(src_width / opt.count) dst_height = int(dst_width / opt.aspect) + +crop_width = dst_width +crop_height = int(dst_width / opt.aspect * crop_aspect) + +crop_size = (crop_width, crop_height,) dst_size = (dst_width, dst_height,) y0 = opt.vertical_offset - dst_height / 2 - margin @@ -46,7 +52,7 @@ for i in range(opt.count): x = int(i / opt.count * dst_width) - margin if x < 0: x += dst_width - crops.append((x, y0, x + dst_width + margin + margin, y0 + dst_height + margin + margin,)) + crops.append((x, y0, x + crop_width + margin + margin, y0 + crop_height + margin + margin,)) path = os.path.join(out_path, chr(97 + i)) os.makedirs(path) paths.append(path) @@ -72,7 +78,7 @@ def build_thumbnail(i, fn): canvas.paste(image, (src_width, 0)) for n, path in enumerate(paths): - image.crop(crops[n]).save(os.path.join(path, out_fn)) + image.crop(crops[n]).resize(dst_size).save(os.path.join(path, out_fn)) chunksize = 3 with Pool(processes=cpu_count()) as pool: |