From 8b7f864ffae2bc57068d230f16a8d568e6963300 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 21 May 2018 04:06:34 +0200 Subject: un rato --- live-mogrify.py | 2 ++ options/dataset_options.py | 6 ++++++ 2 files changed, 8 insertions(+) diff --git a/live-mogrify.py b/live-mogrify.py index 97e7aeb..f218f37 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -222,6 +222,8 @@ def process_live_input(opt, data_opt, rpc_client): cv2.imwrite(tmp_path, next_img) os.rename(tmp_path, next_path) + if data_opt.exit: + sys.exit(0) if __name__ == '__main__': listener = Listener() diff --git a/options/dataset_options.py b/options/dataset_options.py index 543d776..c6530d0 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -93,6 +93,12 @@ class DatasetOptions(BaseOptions): help='after generating a B image, save it to disk' ) + self.parser.add_argument( + '--exit', + action='store_true', + help='exit immediately if set to true (used to interrupt)' + ) + ## IMAGE FILTERS ### RECURSION -- cgit v1.2.3-70-g09d2