diff options
Diffstat (limited to 'test-mogrify.py')
| -rw-r--r-- | test-mogrify.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/test-mogrify.py b/test-mogrify.py index 8ad85a4..8e5c393 100644 --- a/test-mogrify.py +++ b/test-mogrify.py @@ -66,6 +66,8 @@ if __name__ == '__main__': if A_offset: print(">> starting offset: {}".format(A_offset)) A_dir = opt.start_img.replace(numz[0], "{:05d}") + else: + print("Sequence not found") def process_image(im): img = im[:, :, ::-1].copy() @@ -117,7 +119,8 @@ if __name__ == '__main__': tmp_path = render_dir + "frame_{:05d}_tmp.png".format(i+1) edges_path = render_dir + "frame_{:05d}.png".format(i+1) render_path = render_dir + "ren_{:05d}.png".format(i+1) - sequence_path = A_dir.format(A_offset+i+1) + if A_dir is not None: + sequence_path = A_dir.format(A_offset+i+1) # A_offset # save rendered image @@ -129,7 +132,7 @@ if __name__ == '__main__': if data_opt.recursive and last_im is not None: tmp_im = im.copy() - if data_opt.sequence: + if data_opt.sequence and A_dir is not None: A_img = Image.open(sequence_path).convert('RGB') A_im = np.asarray(A_img) frac_a = data_opt.recursive_frac |