oskarastrom commited on
Commit
72ab76e
·
1 Parent(s): 925e44c

Update associative.py

Browse files
Files changed (1) hide show
  1. lib/fish_eye/associative.py +3 -3
lib/fish_eye/associative.py CHANGED
@@ -221,11 +221,11 @@ class Associate(object):
221
 
222
  print("tracks")
223
  print(trks)
224
- high_matched, unmatched_high_dets, unmatched_trks = associate_detections_to_trackers(high_dets, trks, self.iou_threshold)
225
 
226
- print(unmatched_trks)
227
 
228
- low_matched, unmatched_low_dets, unmatched_trks = associate_detections_to_trackers(low_dets, unmatched_trks, self.iou_threshold)
229
 
230
  # update matched trackers with assigned detections
231
  for m in high_matched:
 
221
 
222
  print("tracks")
223
  print(trks)
224
+ high_matched, unmatched_high_dets, unmatched_trk_ids = associate_detections_to_trackers(high_dets, trks, self.iou_threshold)
225
 
226
+ print(trks[unmatched_trks_ids])
227
 
228
+ low_matched, unmatched_low_dets, unmatched_trks_ids = associate_detections_to_trackers(low_dets, trks[unmatched_trks_ids], self.iou_threshold)
229
 
230
  # update matched trackers with assigned detections
231
  for m in high_matched: