diff --git a/ppdet/metrics/metrics.py b/ppdet/metrics/metrics.py index 7f4f5f1fbdffab1d434488bcff13646da95695a8..b473509599b9fbadb48ed792c5100857ab1c30ca 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)