From aa028bde18dab32eba630df45b25ffea840f6ffa Mon Sep 17 00:00:00 2001 From: Atlantisming
Date: Sat, 2 Mar 2024 16:54:00 +0800 Subject: [PATCH] fix dota metric score. --- mmrotate/evaluation/metrics/dota_metric.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmrotate/evaluation/metrics/dota_metric.py b/mmrotate/evaluation/metrics/dota_metric.py index bd4564a4d..410b0939d 100644 --- a/mmrotate/evaluation/metrics/dota_metric.py +++ b/mmrotate/evaluation/metrics/dota_metric.py @@ -197,7 +197,7 @@ def merge_results(self, results: Sequence[dict], else: raise NotImplementedError for qbox, score in zip(qboxes, scores): - txt_element = [img_id, str(round(float(score), 2)) + txt_element = [img_id, str(float(score)) ] + [f'{p:.2f}' for p in qbox] f.writelines(' '.join(txt_element) + '\n')