Merge pull request open-mmlab#885 from SCZwangxiao/master

[fix] Correct the key name of ‘eval_results’ dictionary for metric 'mmit_mean_average_precision'
sibozhang · May 25, 2021 · ae5fb59 · ae5fb59
2 parents 867b022 + 90e50e0
commit ae5fb59
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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