oskarastrom commited on
Commit
5165dd5
1 Parent(s): fa2b33e

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -361,7 +361,7 @@ def do_associative_tracking(inference, image_shapes, width, height, image_meter_
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)
366
  else:
367
  print("empty")
 
361
  low_boxes = low_preds[key]
362
  high_boxes = high_preds[key]
363
  boxes = (low_boxes, high_boxes)
364
+ if low_boxes is not None and high_boxes is not None:
365
  tracker.update(boxes)
366
  else:
367
  print("empty")