Skip to content

Commit

Permalink
Update posetrack_reid.py
Browse files Browse the repository at this point in the history
  • Loading branch information
anDoer committed Mar 28, 2022
1 parent 1bb05ee commit 1fc5de2
Showing 1 changed file with 0 additions and 12 deletions.
12 changes: 0 additions & 12 deletions eval/posetrack21/posetrack21/trackeval/datasets/posetrack_reid.py
Original file line number Diff line number Diff line change
Expand Up @@ -112,7 +112,6 @@ def get_display_name(self, tracker):

def _get_seq_info(self):
sequence_files = os.listdir(self.gt_fol)
# sequence_files = [file for file in sequence_files if file == '009475_mpii_test.json']
seq_lengths = dict()

# reading sequence lengths
Expand Down Expand Up @@ -326,14 +325,6 @@ def get_preprocessed_seq_data(self, raw_data, cls):
num_tracker_dets += len(data['tracker_ids'][t])
num_gt_dets += len(data['gt_ids'][t])

# re-label ids such that there are no empty ids
# todo: this is potentially problematic as we want to consider global assignment over all sequences!!!!!
# this has to be adapted
# current solution: store original ids for later processing and evaluation
###########################################################################
############ this part needs to be adapted!!! otherwise,
############ global re-id does not work
###########################################################################
if len(unique_gt_ids) > 0:
unique_gt_ids = np.unique(unique_gt_ids)
gt_id_map = np.nan * np.ones((np.max(unique_gt_ids) + 1))
Expand Down Expand Up @@ -561,9 +552,6 @@ def get_raw_seq_data(self, tracker, seq):
keypoint_distances = []
keypoint_matches = []
for t, (gt_dets_t, tracker_dets_t, head_sizes_t) in enumerate(zip(raw_data['gt_dets'], raw_data['tracker_dets'], raw_data['head_sizes'])):
# ToDO: return assignment information:
# match label: True/False whether det joint was matched to gt joint
# det label: True/False whether a joint was detected!

pckhs, distances, matches = self._calculate_p_similarities(gt_dets_t, tracker_dets_t, head_sizes_t)
similarity_scores.append(pckhs)
Expand Down

0 comments on commit 1fc5de2

Please sign in to comment.