From 90e50e0f73931971c8f1b51c8a8c1b351ca61664 Mon Sep 17 00:00:00 2001 From: SCZwangxiao Date: Tue, 25 May 2021 17:51:52 +0800 Subject: [PATCH] =?UTF-8?q?Correct=20dictionary=20=E2=80=98eval=5Fresults?= =?UTF-8?q?=E2=80=99=20key=20name=20for=20metric=20mmit=5Fmean=5Faverage?= =?UTF-8?q?=5Fprecision?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- mmaction/datasets/base.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mmaction/datasets/base.py b/mmaction/datasets/base.py index bbed8addec..62fe34f214 100644 --- a/mmaction/datasets/base.py +++ b/mmaction/datasets/base.py @@ -229,10 +229,12 @@ def evaluate(self, ] if metric == 'mean_average_precision': mAP = mean_average_precision(results, gt_labels) + eval_results['mean_average_precision'] = mAP + log_msg = f'\nmean_average_precision\t{mAP:.4f}' elif metric == 'mmit_mean_average_precision': mAP = mmit_mean_average_precision(results, gt_labels) - eval_results['mean_average_precision'] = mAP - log_msg = f'\nmean_average_precision\t{mAP:.4f}' + eval_results['mmit_mean_average_precision'] = mAP + log_msg = f'\nmmit_mean_average_precision\t{mAP:.4f}' print_log(log_msg, logger=logger) continue