oskarastrom commited on
Commit
36f230c
·
1 Parent(s): 0d4fbd4

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -362,8 +362,8 @@ def do_associative_tracking(raw_detections, image_meter_width, image_meter_heigh
362
  with tqdm(total=len(low_dets), desc="Running tracking", ncols=0, disable=not verbose) as pbar:
363
  for i in range(len(low_dets)):
364
  if gp: gp(i / len(low_dets), pbar.__str__())
365
- low_boxes = low_dets[key]
366
- high_boxes = high_dets[key]
367
  boxes = (low_boxes, high_boxes)
368
  if len(low_boxes) + len(high_boxes) > 0:
369
  tracker.update(boxes)
 
362
  with tqdm(total=len(low_dets), desc="Running tracking", ncols=0, disable=not verbose) as pbar:
363
  for i in range(len(low_dets)):
364
  if gp: gp(i / len(low_dets), pbar.__str__())
365
+ low_boxes = low_dets[i]
366
+ high_boxes = high_dets[i]
367
  boxes = (low_boxes, high_boxes)
368
  if len(low_boxes) + len(high_boxes) > 0:
369
  tracker.update(boxes)