modify yolo_video usage description(qqwweee#106)

ytianxing · Jun 12, 2018 · 1cf9ec8 · 1cf9ec8
1 parent 998d70d
commit 1cf9ec8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -18,7 +18,7 @@ A Keras implementation of YOLOv3 (Tensorflow backend) inspired by [allanzelener/
 ```
 wget https://pjreddie.com/media/files/yolov3.weights
 python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
-python yolo.py   OR   python yolo_video.py
+python yolo.py   OR   python yolo_video.py [video_path] [output_path(optional)]
 ```
 
 For Tiny YOLOv3, just do in a similar way. And modify model path and anchor path in yolo.py.

diff --git a/yolo_video.py b/yolo_video.py
@@ -1,7 +1,7 @@
 import sys
 
 if len(sys.argv) < 2:
-    print("Usage: $ python {0} [video_path, [, output_path]]", sys.argv[0])
+    print("Usage: $ python {0} [video_path] [output_path(optional)]", sys.argv[0])
     exit()
 
 from yolo import YOLO
@@ -11,4 +11,6 @@
     video_path = sys.argv[1]
     if len(sys.argv) > 2:
         output_path = sys.argv[2]
-    detect_video(YOLO(), video_path, output_path)
+        detect_video(YOLO(), video_path, output_path)
+    else:
+        detect_video(YOLO(), video_path)