Skip to content

Commit

Permalink
Refactor and comments.
Browse files Browse the repository at this point in the history
  • Loading branch information
CaptainEven committed Nov 6, 2020
1 parent 1a7e672 commit 24f2efb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/lib/tracker/multitracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -659,11 +659,11 @@ def update_tracking(self, im_blob, img_0):
tracked_tracks_dict[cls_id].append(track)

''' Step 2: First association, with embedding'''
# building tracking pool for the current frame
track_pool_dict = defaultdict(list)
track_pool_dict[cls_id] = join_tracks(tracked_tracks_dict[cls_id], self.lost_tracks_dict[cls_id])

# Predict the current location with KF
# for track in track_pool:
Track.multi_predict(track_pool_dict[cls_id])
dists = matching.embedding_distance(track_pool_dict[cls_id], cls_detects)
dists = matching.fuse_motion(self.kalman_filter, dists, track_pool_dict[cls_id], cls_detects)
Expand Down

0 comments on commit 24f2efb

Please sign in to comment.