diff options
| -rw-r--r-- | crop-thirds.py | 6 |
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): |