diff options
| -rw-r--r-- | live-mogrify.py | 2 | ||||
| -rw-r--r-- | options/dataset_options.py | 6 |
2 files changed, 8 insertions, 0 deletions
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 |