Update inference.py

inference.py

@@ -338,17 +338,19 @@ def do_associative_tracking(raw_detections, image_meter_width, image_meter_heigh
 
     low_dets = []
     high_dets = []
-    for batch in raw_detections:
-        for frame in batch:
-            low_frame = []
-            high_frame = []
-            for bbox in frame:
-                if bbox[4] > conf_thresh:
-                    high_frame.append(bbox)
-                else:
-                    low_frame.append(bbox)
-            low_dets.append(low_frame)
-            high_dets.append(high_frame)
+    with tqdm(total=len(raw_detections), desc="Running tracking", ncols=0, disable=not verbose) as pbar:
+        for batch in raw_detections:
+            for frame in batch:
+                low_frame = []
+                high_frame = []
+                for bbox in frame:
+                    if bbox[4] > conf_thresh:
+                        high_frame.append(bbox)
+                    else:
+                        low_frame.append(bbox)
+                low_dets.append(low_frame)
+                high_dets.append(high_frame)
+            pbar.update(1)
 
     print("Preprocess done")