oskarastrom commited on
Commit
e756ca3
·
1 Parent(s): 2f9c05c

Update associative.py

Browse files
Files changed (1) hide show
  1. lib/fish_eye/associative.py +6 -5
lib/fish_eye/associative.py CHANGED
@@ -223,16 +223,17 @@ class Associate(object):
223
  print(trks)
224
  high_matched, unmatched_high_dets, unmatched_trk_ids = associate_detections_to_trackers(high_dets, trks, self.iou_threshold)
225
 
 
 
 
 
226
  if len(unmatched_trk_ids) > 0:
227
  print(trks[unmatched_trk_ids])
228
 
229
  low_matched, unmatched_low_dets, unmatched_trk_ids = associate_detections_to_trackers(low_dets, trks[unmatched_trk_ids], self.iou_threshold)
230
 
231
- # update matched trackers with assigned detections
232
- for m in high_matched:
233
- self.trackers[m[1]].update(high_dets[m[0], :])
234
- for m in low_matched:
235
- self.trackers[m[1]].update(low_dets[m[0], :])
236
 
237
  # create and initialise new trackers for unmatched detections
238
  for i in unmatched_high_dets:
 
223
  print(trks)
224
  high_matched, unmatched_high_dets, unmatched_trk_ids = associate_detections_to_trackers(high_dets, trks, self.iou_threshold)
225
 
226
+ # update matched trackers with assigned detections
227
+ for m in high_matched:
228
+ self.trackers[m[1]].update(high_dets[m[0], :])
229
+
230
  if len(unmatched_trk_ids) > 0:
231
  print(trks[unmatched_trk_ids])
232
 
233
  low_matched, unmatched_low_dets, unmatched_trk_ids = associate_detections_to_trackers(low_dets, trks[unmatched_trk_ids], self.iou_threshold)
234
 
235
+ for m in low_matched:
236
+ self.trackers[m[1]].update(low_dets[m[0], :])
 
 
 
237
 
238
  # create and initialise new trackers for unmatched detections
239
  for i in unmatched_high_dets: