Fixed the display flag (#142)

* Fixed the display flag * Update deep_sort_app.py
nwojke · Apr 1, 2019 · 280b8bd · 280b8bd
1 parent 53ee00b
commit 280b8bd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/deep_sort_app.py b/deep_sort_app.py
@@ -212,6 +212,12 @@ def frame_callback(vis, frame_idx):
             row[0], row[1], row[2], row[3], row[4], row[5]),file=f)
 
 
+def bool_string(input_string):
+    if input_string not in {"True","False"}:
+        raise ValueError("Please Enter a valid Ture/False choice")
+    else:
+        return (input_string == "True")
+
 def parse_args():
     """ Parse command line arguments.
     """
@@ -245,7 +251,7 @@ def parse_args():
         "gallery. If None, no budget is enforced.", type=int, default=None)
     parser.add_argument(
         "--display", help="Show intermediate tracking results",
-        default=True, type=bool)
+        default=True, type=bool_string)
     return parser.parse_args()