From 792cb26924f8039c0310fad1ae6f23a40415c070 Mon Sep 17 00:00:00 2001 From: Tong Gao Date: Thu, 11 Aug 2022 18:00:15 +0800 Subject: [PATCH] [Enhancement] Remove reduntant code snippet in F1Metric (#1278) --- mmocr/evaluation/metrics/f_metric.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/mmocr/evaluation/metrics/f_metric.py b/mmocr/evaluation/metrics/f_metric.py index 28d52ed1..4c4bba1a 100644 --- a/mmocr/evaluation/metrics/f_metric.py +++ b/mmocr/evaluation/metrics/f_metric.py @@ -62,6 +62,7 @@ class F1Metric(BaseMetric): assert isinstance(mode, (list, str)) assert not (len(cared_classes) > 0 and len(ignored_classes) > 0), \ 'cared_classes and ignored_classes cannot be both non-empty' + if isinstance(mode, str): mode = [mode] assert set(mode).issubset({'micro', 'macro'}) @@ -82,9 +83,6 @@ class F1Metric(BaseMetric): self.cared_labels = list(range(num_classes)) self.num_classes = num_classes self.key = key - if isinstance(mode, str): - mode = [mode] - self.mode = mode def process(self, data_batch: Sequence[Dict], predictions: Sequence[Dict]) -> None: