未验证 提交 45ba40b4 编写于 作者: W wangxinxin08 提交者: GitHub

fix dota category while save results (#7473)

上级 ca227ee4
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册