From 45ba40b4dc61270d17885d2ae407dea96b136ee1 Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Thu, 8 Dec 2022 16:44:36 +0800
Subject: [PATCH] fix dota category while save results (#7473)

---
 ppdet/metrics/metrics.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppdet/metrics/metrics.py b/ppdet/metrics/metrics.py
index 7f4f5f1fb..b47350959 100644
--- a/ppdet/metrics/metrics.py
+++ b/ppdet/metrics/metrics.py
@@ -350,7 +350,7 @@ class WiderFaceMetric(Metric):
 class RBoxMetric(Metric):
     def __init__(self, anno_file, **kwargs):
         self.anno_file = anno_file
-        self.clsid2catid, self.catid2name = get_categories('COCO', anno_file)
+        self.clsid2catid, self.catid2name = get_categories('RBOX', anno_file)
         self.catid2clsid = {v: k for k, v in self.clsid2catid.items()}
         self.classwise = kwargs.get('classwise', False)
         self.output_eval = kwargs.get('output_eval', None)
-- 
GitLab