go hey

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