crop-thirds

1 files changed, 9 insertions, 3 deletions

diff --git a/crop-equirectangular.py b/crop-equirectangular.py
index 38e93ae..317c34e 100644
--- a/crop-equirectangular.py
+++ b/crop-equirectangular.py
@@ -17,6 +17,8 @@ parser = argparse.ArgumentParser()
 parser.add_argument('--folder', default="./sequences/venice_360/equi")
 parser.add_argument('--label', required=True)
 parser.add_argument('--vertical_offset', type=int, default=983)
+# parser.add_argument('--dst_width', type=int, default=1024)
+# parser.add_argument('--dst_height', type=int, default=512)
 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')
@@ -25,6 +27,10 @@ opt = parser.parse_args()
 src_width = 4096
 src_height = 2048
 
+dst_width = 1024
+dst_height = 512
+dst_size = (dst_width, dst_height,)
+
 count = 3
 output_aspect = 3
 
@@ -86,9 +92,9 @@ def build_thumbnail(i, fn):
   canvas.paste(image, (0, 0))
   canvas.paste(image, (src_width, 0))
 
-  canvas.crop(p0).resize((1024, 512,), Image.ANTIALIAS).save(os.path.join(path_0, out_fn))
-  canvas.crop(p1).resize((1024, 512,), Image.ANTIALIAS).save(os.path.join(path_1, out_fn))
-  canvas.crop(p2).resize((1024, 512,), Image.ANTIALIAS).save(os.path.join(path_2, out_fn))
+  canvas.crop(p0).resize(dst_size, Image.ANTIALIAS).save(os.path.join(path_0, out_fn))
+  canvas.crop(p1).resize(dst_size, Image.ANTIALIAS).save(os.path.join(path_1, out_fn))
+  canvas.crop(p2).resize(dst_size, Image.ANTIALIAS).save(os.path.join(path_2, out_fn))
 
 chunksize = 3
 with Pool(processes=cpu_count()) as pool: