Skip to content

Commit 443e979

Browse files
KevinNuNuKevinNuNu
committed
优化xfund数据集的config_generator命名，使config_generator目录结构更清晰
1 parent deb96cc commit 443e979

File tree

20 files changed

+208
-213
lines changed

20 files changed

+208
-213
lines changed

‎configs/re/_base_/datasets/xfund_zh.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
xfund_zh_re_data_root = 'data/xfund/zh'
22

33
xfund_zh_re_train = dict(
4-
type='REDataset',
4+
type='XFUNDREDataset',
55
data_root=xfund_zh_re_data_root,
66
ann_file='re_train.json',
77
pipeline=None)
88

99
xfund_zh_re_test = dict(
10-
type='REDataset',
10+
type='XFUNDREDataset',
1111
data_root=xfund_zh_re_data_root,
1212
ann_file='re_test.json',
1313
test_mode=True,

‎configs/ser/_base_/datasets/xfund_zh.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
xfund_zh_ser_data_root = 'data/xfund/zh'
22

33
xfund_zh_ser_train = dict(
4-
type='SERDataset',
4+
type='XFUNDSERDataset',
55
data_root=xfund_zh_ser_data_root,
66
ann_file='ser_train.json',
77
pipeline=None)
88

99
xfund_zh_ser_test = dict(
10-
type='SERDataset',
10+
type='XFUNDSERDataset',
1111
data_root=xfund_zh_ser_data_root,
1212
ann_file='ser_test.json',
1313
test_mode=True,

‎dataset_zoo/xfund/de/re.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33
_base_.train_preparer.packer.type = 'REPacker'
44
_base_.test_preparer.packer.type = 'REPacker'
55

6-
config_generator = dict(type='REConfigGenerator')
6+
config_generator = dict(type='XFUNDREConfigGenerator')

‎dataset_zoo/xfund/de/ser.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -57,4 +57,4 @@
5757
)
5858

5959
delete = ['annotations'] + [f'{lang}_{split}' for split in ['train', 'val']]
60-
config_generator = dict(type='SERConfigGenerator')
60+
config_generator = dict(type='XFUNDSERConfigGenerator')

‎dataset_zoo/xfund/es/re.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33
_base_.train_preparer.packer.type = 'REPacker'
44
_base_.test_preparer.packer.type = 'REPacker'
55

6-
config_generator = dict(type='REConfigGenerator')
6+
config_generator = dict(type='XFUNDREConfigGenerator')

‎dataset_zoo/xfund/es/ser.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -57,4 +57,4 @@
5757
)
5858

5959
delete = ['annotations'] + [f'{lang}_{split}' for split in ['train', 'val']]
60-
config_generator = dict(type='SERConfigGenerator')
60+
config_generator = dict(type='XFUNDSERConfigGenerator')

‎dataset_zoo/xfund/fr/re.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33
_base_.train_preparer.packer.type = 'REPacker'
44
_base_.test_preparer.packer.type = 'REPacker'
55

6-
config_generator = dict(type='REConfigGenerator')
6+
config_generator = dict(type='XFUNDREConfigGenerator')

‎dataset_zoo/xfund/fr/ser.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -57,4 +57,4 @@
5757
)
5858

5959
delete = ['annotations'] + [f'{lang}_{split}' for split in ['train', 'val']]
60-
config_generator = dict(type='SERConfigGenerator')
60+
config_generator = dict(type='XFUNDSERConfigGenerator')

‎dataset_zoo/xfund/it/re.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33
_base_.train_preparer.packer.type = 'REPacker'
44
_base_.test_preparer.packer.type = 'REPacker'
55

6-
config_generator = dict(type='REConfigGenerator')
6+
config_generator = dict(type='XFUNDREConfigGenerator')

‎dataset_zoo/xfund/it/ser.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -57,4 +57,4 @@
5757
)
5858

5959
delete = ['annotations'] + [f'{lang}_{split}' for split in ['train', 'val']]
60-
config_generator = dict(type='SERConfigGenerator')
60+
config_generator = dict(type='XFUNDSERConfigGenerator')

‎dataset_zoo/xfund/ja/re.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33
_base_.train_preparer.packer.type = 'REPacker'
44
_base_.test_preparer.packer.type = 'REPacker'
55

6-
config_generator = dict(type='REConfigGenerator')
6+
config_generator = dict(type='XFUNDREConfigGenerator')

‎dataset_zoo/xfund/ja/ser.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -57,4 +57,4 @@
5757
)
5858

5959
delete = ['annotations'] + [f'{lang}_{split}' for split in ['train', 'val']]
60-
config_generator = dict(type='SERConfigGenerator')
60+
config_generator = dict(type='XFUNDSERConfigGenerator')

‎dataset_zoo/xfund/pt/re.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33
_base_.train_preparer.packer.type = 'REPacker'
44
_base_.test_preparer.packer.type = 'REPacker'
55

6-
config_generator = dict(type='REConfigGenerator')
6+
config_generator = dict(type='XFUNDREConfigGenerator')

‎dataset_zoo/xfund/pt/ser.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -57,4 +57,4 @@
5757
)
5858

5959
delete = ['annotations'] + [f'{lang}_{split}' for split in ['train', 'val']]
60-
config_generator = dict(type='SERConfigGenerator')
60+
config_generator = dict(type='XFUNDSERConfigGenerator')

‎dataset_zoo/xfund/zh/re.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33
_base_.train_preparer.packer.type = 'REPacker'
44
_base_.test_preparer.packer.type = 'REPacker'
55

6-
config_generator = dict(type='REConfigGenerator')
6+
config_generator = dict(type='XFUNDREConfigGenerator')

‎dataset_zoo/xfund/zh/ser.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -57,4 +57,4 @@
5757
)
5858

5959
delete = ['annotations'] + [f'{lang}_{split}' for split in ['train', 'val']]
60-
config_generator = dict(type='SERConfigGenerator')
60+
config_generator = dict(type='XFUNDSERConfigGenerator')

‎mmocr/datasets/preparers/config_generators/__init__.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
# Copyright (c) OpenMMLab. All rights reserved.
22
from .base import BaseDatasetConfigGenerator
3-
from .re_config_generator import REConfigGenerator
4-
from .ser_config_generator import SERConfigGenerator
53
from .textdet_config_generator import TextDetConfigGenerator
64
from .textrecog_config_generator import TextRecogConfigGenerator
75
from .textspotting_config_generator import TextSpottingConfigGenerator
6+
from .xfund_config_generator import (XFUNDREConfigGenerator,
7+
XFUNDSERConfigGenerator)
88

99
__all__ = [
1010
'BaseDatasetConfigGenerator', 'TextDetConfigGenerator',
1111
'TextRecogConfigGenerator', 'TextSpottingConfigGenerator',
12-
'SERConfigGenerator', 'REConfigGenerator'
12+
'XFUNDSERConfigGenerator', 'XFUNDREConfigGenerator'
1313
]

‎mmocr/datasets/preparers/config_generators/re_config_generator.py

-96
This file was deleted.

‎mmocr/datasets/preparers/config_generators/ser_config_generator.py

-96
This file was deleted.

0 commit comments

Comments
 (0)