Skip to content

Commit a61c92f

Browse files
allow no distractor classes, do not deafult to MOT17s
1 parent 8ae1482 commit a61c92f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

boxmot/utils/custom_mot_challenge_2d_box.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -66,10 +66,10 @@ def get_preprocessed_seq_data(self, raw_data, cls):
6666
self._check_unique_ids(raw_data)
6767
cls_id = self.class_name_to_class_id[cls]
6868
# MOT distractor set; prefer config override, otherwise fall back to legacy defaults
69-
if self.distractor_class_ids is not None:
70-
distractor_classes = self.distractor_class_ids
71-
else:
72-
distractor_classes = [12, 8, 6, 7, 2] if self.benchmark == 'MOT20' else [12, 8, 7, 2]
69+
#if self.distractor_class_ids is not None:
70+
distractor_classes = self.distractor_class_ids
71+
# else:
72+
# distractor_classes = [12, 8, 6, 7, 2] if self.benchmark == 'MOT20' else [12, 8, 7, 2]
7373

7474
data_keys = ['gt_ids', 'tracker_ids', 'gt_dets', 'tracker_dets',
7575
'tracker_confidences', 'similarity_scores']

0 commit comments

Comments
 (0)