Spaces:
Runtime error
Runtime error
Commit
·
e756ca3
1
Parent(s):
2f9c05c
Update associative.py
Browse files
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 |
-
|
232 |
-
|
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:
|