projectstreet-detection-docker

commit 7167913f539dde666ce911196826a099a7897e3e
parent 27f53c446fe7dbf83c962bba291d38a9ffe3bdf5
Author: Marcel <MTRNord@users.noreply.github.com>
Date:   Sun,  9 Oct 2016 16:28:47 +0200

Update stage3_analyseVideo.py
Diffstat:
M
scripts/stage3_analyseVideo.py
 | 
13
+++++++++++--

1 file changed, 11 insertions(+), 2 deletions(-)
diff --git a/scripts/stage3_analyseVideo.py b/scripts/stage3_analyseVideo.py
@@ -9,10 +9,19 @@
 ##### Video to single frames comes from: http://stackoverflow.com/a/19082750/4929236 #####
 
 import cv2
+import optparse
 
-cap = cv2.VideoCapture("./out.mp4")
+parser = optparse.OptionParser()
+
+parser.add_option('-v', '--video',
+    action="store", dest="video",
+    help="Video to analyse", default="./in.mp4")
+
+options, args = parser.parse_args()
+
+cap = cv2.VideoCapture(options.video)
 while not cap.isOpened():
-    cap = cv2.VideoCapture("./out.mp4")
+    cap = cv2.VideoCapture(options.video)
     cv2.waitKey(1000)
     print "Wait for the header"