From bd21e5c16fbae5c29140adc1c9eb47e57a0e333a Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Tue, 15 May 2018 05:26:12 +0200
Subject: testy

---
 test-mogrify.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'test-mogrify.py')

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()
-- 
cgit v1.2.3-70-g09d2