Spaces:
Runtime error
Runtime error
Commit ·
fa2b33e
1
Parent(s): 3986029
Update inference.py
Browse files- inference.py +3 -3
inference.py
CHANGED
|
@@ -356,10 +356,10 @@ def do_associative_tracking(inference, image_shapes, width, height, image_meter_
|
|
| 356 |
|
| 357 |
# Run tracking
|
| 358 |
with tqdm(total=len(low_preds), desc="Running tracking", ncols=0, disable=not verbose) as pbar:
|
| 359 |
-
for i in
|
| 360 |
if gp: gp(i / len(low_preds), pbar.__str__())
|
| 361 |
-
low_boxes = low_preds[
|
| 362 |
-
high_boxes = high_preds[
|
| 363 |
boxes = (low_boxes, high_boxes)
|
| 364 |
if len(low_boxes) + len(high_boxes) > 0:
|
| 365 |
tracker.update(boxes)
|
|
|
|
| 356 |
|
| 357 |
# Run tracking
|
| 358 |
with tqdm(total=len(low_preds), desc="Running tracking", ncols=0, disable=not verbose) as pbar:
|
| 359 |
+
for i, key in enumerate(sorted(low_preds.keys())):
|
| 360 |
if gp: gp(i / len(low_preds), pbar.__str__())
|
| 361 |
+
low_boxes = low_preds[key]
|
| 362 |
+
high_boxes = high_preds[key]
|
| 363 |
boxes = (low_boxes, high_boxes)
|
| 364 |
if len(low_boxes) + len(high_boxes) > 0:
|
| 365 |
tracker.update(boxes)
|