testy

1 files changed, 2 insertions, 2 deletions

diff --git a/test-mogrify.py b/test-mogrify.py
index c6deaa3..083154a 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -66,7 +66,7 @@ if __name__ == '__main__':
         print(A_dir)
       else:
         print("Sequence not found")
-    sys.exit(1)
+    return A_offset, A_im, A_dir
 
   def process_image(im):
     img = im[:, :, ::-1].copy()
@@ -153,7 +153,7 @@ if __name__ == '__main__':
     frac_b = 1.0 - frac_a
     print("rec: {}, pix: {}".format(frac_a, frac_b))
 
-  load_first_frame()
+  A_offset, A_im, A_dir = load_first_frame()
 
   data_loader = CreateRecursiveDataLoader(opt)
   dataset = data_loader.load_data()