From 386d1eadfeb900cdd990fbac66aeb1428553a5f7 Mon Sep 17 00:00:00 2001 From: George Ni Date: Wed, 7 Jul 2021 11:36:58 +0800 Subject: [PATCH] MOT] fix read_mot_results (#3587) --- ppdet/metrics/mot_eval_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppdet/metrics/mot_eval_utils.py b/ppdet/metrics/mot_eval_utils.py index 26b3911a5..8b0a03a13 100644 --- a/ppdet/metrics/mot_eval_utils.py +++ b/ppdet/metrics/mot_eval_utils.py @@ -43,14 +43,14 @@ def read_mot_results(filename, is_gt=False, is_ignore=False): box_size = float(linelist[4]) * float(linelist[5]) if is_gt: - if 'MOT16-' in filename or 'MOT17-' in filename: + if 'MOT16-' in filename or 'MOT17-' in filename or 'MOT15-' in filename or 'MOT20-' in filename: label = int(float(linelist[7])) mark = int(float(linelist[6])) if mark == 0 or label not in valid_labels: continue score = 1 elif is_ignore: - if 'MOT16-' in filename or 'MOT17-' in filename: + if 'MOT16-' in filename or 'MOT17-' in filename or 'MOT15-' in filename or 'MOT20-' in filename: label = int(float(linelist[7])) vis_ratio = float(linelist[8]) if label not in ignore_labels and vis_ratio >= 0: -- GitLab