summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--test-mogrify.py4
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()