From 22a8b9f5bbd6a953207c2b5a894bcd9ac4b17daf Mon Sep 17 00:00:00 2001 From: RangiLyu Date: Thu, 13 Oct 2022 11:38:35 +0800 Subject: [PATCH] [Enhance] Log coco metric copypaste. --- mmdet/evaluation/metrics/coco_metric.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mmdet/evaluation/metrics/coco_metric.py b/mmdet/evaluation/metrics/coco_metric.py index 7c1f1855d9f..c8f101402e8 100644 --- a/mmdet/evaluation/metrics/coco_metric.py +++ b/mmdet/evaluation/metrics/coco_metric.py @@ -531,6 +531,11 @@ def compute_metrics(self, results: list) -> Dict[str, float]: val = coco_eval.stats[coco_metric_names[metric_item]] eval_results[key] = float(f'{round(val, 3)}') + ap = coco_eval.stats[:6] + logger.info(f'{metric}_mAP_copypaste: {ap[0]:.3f} ' + f'{ap[1]:.3f} {ap[2]:.3f} {ap[3]:.3f} ' + f'{ap[4]:.3f} {ap[5]:.3f}') + if tmp_dir is not None: tmp_dir.cleanup() return eval_results