fix bugs in PR #1035

fix tuple unpacking bug in PR #1035

Fix bug with detections with un-initialized tracker_ids
This commit is contained in:
Raif Olson committed 2024-03-25 23:52:33 -04:00
1 parent 97b8312570
commit 3ed4aeffff
1 file changed
+13 -3
+13 -3
View File
@@ -286,13 +286,23 @@ class ByteTrack:
iou_costs = 1 - ious
matches, _, _ = matching.linear_assignment(iou_costs, 0.5)
for i, idet, itrack in enumerate(matches):
for i, (idet, itrack) in enumerate(matches):
if i == 0:
final_detections = detections[[idet]]
final_detections.tracker_id[0] = int(tracks[itrack].track_id)
if final_detections.tracker_id is None:
final_detections.tracker_id = np.asarray(
[int(tracks[itrack].track_id)]
)
else:
final_detections.tracker_id[0] = int(tracks[itrack].track_id)
else:
current_detection = detections[[idet]]
current_detection.tracker_id[0] = int(tracks[itrack].track_id)
if current_detection.tracker_id is None:
current_detection.tracker_id = np.asarray(
[int(tracks[itrack].track_id)]
)
else:
current_detection.tracker_id[0] = int(tracks[itrack].track_id)
final_detections = Detections.merge(
[final_detections, current_detection]
)