testy

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