Tracker only initialized if tracker argument is passed

sovit-123 · Aug 23, 2023 · b417188 · b417188
1 parent e5f8961
commit b417188
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/inference_video.py b/inference_video.py
@@ -109,8 +109,8 @@ def main(args):
     # For same annotation colors each time.
     np.random.seed(42)
 
-    # Initialize Deep SORT tracker if tracker is selected.
-    tracker = DeepSort(max_age=30)
+    if args['track']: # Initialize Deep SORT tracker if tracker is selected.
+        tracker = DeepSort(max_age=30)
 
     # Load the data configurations.
     data_configs = None

diff --git a/onnx_inference_video.py b/onnx_inference_video.py
@@ -105,8 +105,8 @@ def parse_opt():
 
 def main(args):
     np.random.seed(42)
-    # Initialize Deep SORT tracker if tracker is selected.
-    tracker = DeepSort(max_age=30)
+    if args['track']: # Initialize Deep SORT tracker if tracker is selected.
+        tracker = DeepSort(max_age=30)
     # Load model.
     ort_session = onnxruntime.InferenceSession(
         args['weights'], providers=['CUDAExecutionProvider', 'CPUExecutionProvider']