oskarastrom commited on
Commit
4b6e487
·
1 Parent(s): 2d0ef6d

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +5 -0
inference.py CHANGED
@@ -338,6 +338,7 @@ def do_associative_tracking(raw_detections, image_meter_width, image_meter_heigh
338
  high_dets = []
339
  for batch in raw_detections:
340
  for frame in batch:
 
341
  low_frame = []
342
  high_frame = []
343
  for bbox in frame:
@@ -366,8 +367,12 @@ def do_associative_tracking(raw_detections, image_meter_width, image_meter_heigh
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)
370
  else:
 
371
  print(low_boxes)
372
  print(high_boxes)
373
  tracker.update((np.empty((0, 5)), np.empty((0, 5))))
 
338
  high_dets = []
339
  for batch in raw_detections:
340
  for frame in batch:
341
+ print(frame)
342
  low_frame = []
343
  high_frame = []
344
  for bbox in frame:
 
367
  high_boxes = high_dets[i]
368
  boxes = (low_boxes, high_boxes)
369
  if len(low_boxes) + len(high_boxes) > 0:
370
+ print("found")
371
+ print(low_boxes)
372
+ print(high_boxes)
373
  tracker.update(boxes)
374
  else:
375
+ print("empty")
376
  print(low_boxes)
377
  print(high_boxes)
378
  tracker.update((np.empty((0, 5)), np.empty((0, 5))))