thirds

1 files changed, 5 insertions, 1 deletions

diff --git a/crop-thirds.py b/crop-thirds.py
index 1e2b3ab..670dbb2 100644
--- a/crop-thirds.py
+++ b/crop-thirds.py
@@ -21,6 +21,7 @@ parser.add_argument('--vertical_offset', type=int, default=256)
 # parser.add_argument('--dst_width', type=int, default=1024)
 # parser.add_argument('--dst_height', type=int, default=512)
 parser.add_argument('--count', type=int, default=3)
+parser.add_argument('--max', type=int, default=0)
 parser.add_argument('--aspect', type=float, default=3.0)
 # parser.add_argument('--folder_id', type=int, required=True)
 parser.add_argument('--overlap', type=float, default=0.5)
@@ -48,11 +49,14 @@ for i in range(opt.count):
   os.makedirs(path)
   paths.append(path)
 
+max_i = opt.max if opt.max == 0 else 99997
 dataset = []
 for i, fn in enumerate(sorted(glob.glob(os.path.join(opt.folder, '*.png')))):
+  if i > max_i:
+    break
   out_fn = "frame_{:05d}.png".format(i + 1)
   if not opt.clobber and os.path.exists(os.path.join(paths[0], out_fn)):
-     continue
+    continue
   dataset.append((i, fn,))
 
 def build_thumbnail(i, fn):