mirrors/mmocr
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmocr.git synced 2025-06-03 21:54:47 +08:00
Code Issues Projects Releases Wiki Activity

[Enhancement] Remove reduntant code snippet in F1Metric (#1278)

Browse Source
This commit is contained in:
Tong Gao 2022-08-11 18:00:15 +08:00 committed by GitHub
parent 7b25b62c21
commit 792cb26924
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmocr/evaluation/metrics/f_metric.py
View File

@ -62,6 +62,7 @@ class F1Metric(BaseMetric):
assert isinstance(mode, (list, str)) assert isinstance(mode, (list, str))
assert not (len(cared_classes) > 0 and len(ignored_classes) > 0), \ assert not (len(cared_classes) > 0 and len(ignored_classes) > 0), \
'cared_classes and ignored_classes cannot be both non-empty' 'cared_classes and ignored_classes cannot be both non-empty'
if isinstance(mode, str): if isinstance(mode, str):
mode = [mode] mode = [mode]
assert set(mode).issubset({'micro', 'macro'}) assert set(mode).issubset({'micro', 'macro'})
@ -82,9 +83,6 @@ class F1Metric(BaseMetric):
self.cared_labels = list(range(num_classes)) self.cared_labels = list(range(num_classes))
self.num_classes = num_classes self.num_classes = num_classes
self.key = key self.key = key
if isinstance(mode, str):
mode = [mode]
self.mode = mode
def process(self, data_batch: Sequence[Dict], def process(self, data_batch: Sequence[Dict],
predictions: Sequence[Dict]) -> None: predictions: Sequence[Dict]) -> None: