summaryrefslogtreecommitdiff
path: root/run.py
diff options
context:
space:
mode:
Diffstat (limited to 'run.py')
-rw-r--r--run.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/run.py b/run.py
index 744cb9f..aad97a7 100644
--- a/run.py
+++ b/run.py
@@ -16,6 +16,7 @@ arguments_strModel = 'lf'
arguments_strFirst = './images/first.png'
arguments_strSecond = './images/second.png'
arguments_strOut = './result.png'
+arguments_strVideo = False
arguments_strVideoOut = datetime.now().strftime("sepconv_%Y%m%d_%H%M.mp4")
arguments_steps = 0
@@ -32,6 +33,9 @@ for strOption, strArgument in getopt.getopt(sys.argv[1:], '', [ strParameter[2:]
elif strOption == '--out':
arguments_strOut = strArgument # path to where the output should be stored
+ elif strOption == '--video':
+ arguments_strVideoOut = strArgument # path to video
+
elif strOption == '--video-out':
arguments_strVideoOut = strArgument # path to where the video should be stored