From 8bb0ced2580e78dba077aa0b5e0997decb4714a7 Mon Sep 17 00:00:00 2001 From: Alex Bewley Date: Sun, 19 Apr 2020 19:08:19 +0200 Subject: [PATCH] Minor cleanup. --- .gitignore | 1 + sort.py | 3 --- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/.gitignore b/.gitignore index ea1472e..5f7d928 100644 --- a/.gitignore +++ b/.gitignore @@ -1 +1,2 @@ output/ +mot_benchmark/ diff --git a/sort.py b/sort.py index 43d314d..0db046b 100644 --- a/sort.py +++ b/sort.py @@ -162,8 +162,6 @@ def associate_detections_to_trackers(detections,trackers,iou_threshold = 0.3): for t,trk in enumerate(trackers): iou_matrix[d,t] = iou(det,trk) - # TODO (bewley): remove rows and cols iou.max() < threshold - if min(iou_matrix.shape) > 0: a = (iou_matrix > iou_threshold).astype(np.int32) if a.sum(1).max() == 1 and a.sum(0).max() == 1: @@ -287,7 +285,6 @@ if __name__ == '__main__': pattern = os.path.join(args.seq_path, phase, '*', 'det', 'det.txt') for seq_dets_fn in glob.glob(pattern): mot_tracker = Sort() #create instance of the SORT tracker - print(seq_dets_fn) seq_dets = np.loadtxt(seq_dets_fn, delimiter=',') seq = seq_dets_fn[pattern.find('*'):].split('/')[0]