diff options
Diffstat (limited to 'test-mogrify.py')
| -rw-r--r-- | test-mogrify.py | 4 |
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() |