security_vision/sort
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Minor cleanup.

Browse source
This commit is contained in:
Alex Bewley 2020-04-19 19:08:19 +02:00
parent 6be936b01e
commit 8bb0ced258
2 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
.gitignore vendored
View file

@ -1 +1,2 @@
output/ output/
mot_benchmark/

3
sort.py
View file

@ -162,8 +162,6 @@ def associate_detections_to_trackers(detections,trackers,iou_threshold = 0.3):
for t,trk in enumerate(trackers): for t,trk in enumerate(trackers):
iou_matrix[d,t] = iou(det,trk) iou_matrix[d,t] = iou(det,trk)
# TODO (bewley): remove rows and cols iou.max() < threshold
if min(iou_matrix.shape) > 0: if min(iou_matrix.shape) > 0:
a = (iou_matrix > iou_threshold).astype(np.int32) a = (iou_matrix > iou_threshold).astype(np.int32)
if a.sum(1).max() == 1 and a.sum(0).max() == 1: 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') pattern = os.path.join(args.seq_path, phase, '*', 'det', 'det.txt')
for seq_dets_fn in glob.glob(pattern): for seq_dets_fn in glob.glob(pattern):
mot_tracker = Sort() #create instance of the SORT tracker mot_tracker = Sort() #create instance of the SORT tracker
print(seq_dets_fn)
seq_dets = np.loadtxt(seq_dets_fn, delimiter=',') seq_dets = np.loadtxt(seq_dets_fn, delimiter=',')
seq = seq_dets_fn[pattern.find('*'):].split('/')[0] seq = seq_dets_fn[pattern.find('*'):].split('/')[0]