[Enchance] Put all registry into registry.py (#1760)

mmocr/datasets/preparers/config_generators/textdet_config_generator.py

@@ -1,7 +1,7 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 from typing import Dict, List, Optional
 
-from ..data_preparer import CFG_GENERATORS
+from mmocr.registry import CFG_GENERATORS
 from .base import BaseDatasetConfigGenerator
 
 

mmocr/datasets/preparers/config_generators/textrecog_config_generator.py

@@ -1,7 +1,7 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 from typing import Dict, List, Optional
 
-from ..data_preparer import CFG_GENERATORS
+from mmocr.registry import CFG_GENERATORS
 from .base import BaseDatasetConfigGenerator
 
 

mmocr/datasets/preparers/config_generators/textspotting_config_generator.py

@@ -1,7 +1,7 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 from typing import Dict, List, Optional
 
-from ..data_preparer import CFG_GENERATORS
+from mmocr.registry import CFG_GENERATORS
 from .base import BaseDatasetConfigGenerator
 from .textdet_config_generator import TextDetConfigGenerator
 

mmocr/datasets/preparers/data_preparer.py

@@ -5,20 +5,11 @@ import os.path as osp
 import shutil
 from typing import List, Optional, Union
 
-from mmengine import Registry
-
+from mmocr.registry import (CFG_GENERATORS, DATA_DUMPERS, DATA_GATHERERS,
+                            DATA_OBTAINERS, DATA_PACKERS, DATA_PARSERS)
 from mmocr.utils.typing_utils import ConfigType, OptConfigType
 
-DATA_PREPARERS = Registry('data preparer')
-DATA_OBTAINERS = Registry('data_obtainer')
-DATA_GATHERERS = Registry('data_gatherer')
-DATA_PARSERS = Registry('data_parser')
-DATA_DUMPERS = Registry('data_dumper')
-DATA_PACKERS = Registry('data_packer')
-CFG_GENERATORS = Registry('cfg_generator')
 
-
-@DATA_PREPARERS.register_module()
 class DatasetPreparer:
     """Base class of dataset preparer.
 

mmocr/datasets/preparers/dumpers/json_dumper.py

@@ -4,7 +4,7 @@ from typing import Dict
 
 import mmengine
 
-from ..data_preparer import DATA_DUMPERS
+from mmocr.registry import DATA_DUMPERS
 from .base import BaseDumper
 
 

mmocr/datasets/preparers/dumpers/wild_receipt_openset_dumper.py

@@ -2,8 +2,8 @@
 import os.path as osp
 from typing import List
 
+from mmocr.registry import DATA_DUMPERS
 from mmocr.utils import list_to_file
-from ..data_preparer import DATA_DUMPERS
 from .base import BaseDumper
 
 

mmocr/datasets/preparers/gatherers/mono_gatherer.py

@@ -2,7 +2,7 @@
 import os.path as osp
 from typing import Tuple
 
-from ..data_preparer import DATA_GATHERERS
+from mmocr.registry import DATA_GATHERERS
 from .base import BaseGatherer
 
 

mmocr/datasets/preparers/gatherers/naf_gatherer.py

@@ -5,7 +5,7 @@ import os.path as osp
 import shutil
 from typing import List, Tuple
 
-from ..data_preparer import DATA_GATHERERS
+from mmocr.registry import DATA_GATHERERS
 from .base import BaseGatherer
 
 

mmocr/datasets/preparers/gatherers/pair_gatherer.py

@@ -3,8 +3,8 @@ import os.path as osp
 import re
 from typing import List, Optional, Tuple
 
+from mmocr.registry import DATA_GATHERERS
 from mmocr.utils import list_files
-from ..data_preparer import DATA_GATHERERS
 from .base import BaseGatherer
 
 

mmocr/datasets/preparers/obtainers/naive_data_obtainer.py

@@ -9,8 +9,8 @@ from typing import Dict, List, Optional, Tuple
 
 from mmengine import mkdir_or_exist
 
+from mmocr.registry import DATA_OBTAINERS
 from mmocr.utils import check_integrity, is_archive
-from ..data_preparer import DATA_OBTAINERS
 
 ssl._create_default_https_context = ssl._create_unverified_context
 

mmocr/datasets/preparers/packers/textdet_packer.py

@@ -4,8 +4,8 @@ from typing import Dict, List, Tuple
 
 import mmcv
 
+from mmocr.registry import DATA_PACKERS
 from mmocr.utils import bbox2poly, poly2bbox
-from ..data_preparer import DATA_PACKERS
 from .base import BasePacker
 
 

mmocr/datasets/preparers/packers/textrecog_packer.py

@@ -5,8 +5,8 @@ from typing import Dict, List, Tuple
 import mmcv
 from mmengine import mkdir_or_exist
 
+from mmocr.registry import DATA_PACKERS
 from mmocr.utils import bbox2poly, crop_img, poly2bbox, warp_img
-from ..data_preparer import DATA_PACKERS
 from .base import BasePacker
 
 

mmocr/datasets/preparers/packers/textspotting_packer.py

@@ -4,8 +4,8 @@ from typing import Dict, List, Tuple
 
 import mmcv
 
+from mmocr.registry import DATA_PACKERS
 from mmocr.utils import bbox2poly, poly2bbox
-from ..data_preparer import DATA_PACKERS
 from .base import BasePacker
 
 

mmocr/datasets/preparers/packers/wildreceipt_packer.py

@@ -2,7 +2,7 @@
 import json
 from typing import List
 
-from ..data_preparer import DATA_PACKERS
+from mmocr.registry import DATA_PACKERS
 from .base import BasePacker
 
 

mmocr/datasets/preparers/parsers/coco_parser.py

@@ -4,8 +4,8 @@ from typing import List
 
 from mmdet.datasets.api_wrappers import COCO
 
-from mmocr.datasets.preparers.data_preparer import DATA_PARSERS
 from mmocr.datasets.preparers.parsers.base import BaseParser
+from mmocr.registry import DATA_PARSERS
 
 
 @DATA_PARSERS.register_module()

mmocr/datasets/preparers/parsers/funsd_parser.py

@@ -2,8 +2,8 @@
 import json
 from typing import Tuple
 
+from mmocr.registry import DATA_PARSERS
 from mmocr.utils import bbox2poly
-from ..data_preparer import DATA_PARSERS
 from .base import BaseParser
 
 

mmocr/datasets/preparers/parsers/icdar_txt_parser.py

@@ -2,8 +2,8 @@
 import os.path as osp
 from typing import List, Optional, Tuple
 
+from mmocr.registry import DATA_PARSERS
 from mmocr.utils import bbox2poly
-from ..data_preparer import DATA_PARSERS
 from .base import BaseParser
 
 

mmocr/datasets/preparers/parsers/naf_parser.py

@@ -4,7 +4,7 @@ from typing import List, Tuple
 
 import numpy as np
 
-from ..data_preparer import DATA_PARSERS
+from mmocr.registry import DATA_PARSERS
 from .base import BaseParser
 
 

mmocr/datasets/preparers/parsers/sroie_parser.py

@@ -1,8 +1,8 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 from typing import List, Optional, Tuple
 
+from mmocr.registry import DATA_PARSERS
 from mmocr.utils import bbox2poly
-from ..data_preparer import DATA_PARSERS
 from .base import BaseParser
 
 

mmocr/datasets/preparers/parsers/svt_parser.py

@@ -3,7 +3,7 @@ import os.path as osp
 import xml.etree.ElementTree as ET
 from typing import List, Tuple
 
-from ..data_preparer import DATA_PARSERS
+from mmocr.registry import DATA_PARSERS
 from .base import BaseParser
 
 

mmocr/datasets/preparers/parsers/totaltext_parser.py

@@ -4,7 +4,7 @@ from typing import Dict, Tuple
 
 import yaml
 
-from ..data_preparer import DATA_PARSERS
+from mmocr.registry import DATA_PARSERS
 from .base import BaseParser
 
 

mmocr/datasets/preparers/parsers/wildreceipt_parser.py

@@ -3,8 +3,8 @@ import json
 import os.path as osp
 from typing import Dict
 
+from mmocr.registry import DATA_PARSERS
 from mmocr.utils import list_from_file
-from ..data_preparer import DATA_PARSERS
 from .base import BaseParser
 
 

mmocr/registry.py

@@ -132,3 +132,26 @@ LOG_PROCESSORS = Registry(
     parent=MMENGINE_LOG_PROCESSORS,
     # TODO: update the location when mmocr has its own log processor
     locations=['mmocr.engine'])
+# manage data obtainer
+DATA_OBTAINERS = Registry(
+    'data obtainer', locations=['mmocr.datasets.preparers.obtainers'])
+
+# manage data gatherer
+DATA_GATHERERS = Registry(
+    'data gatherer', locations=['mmocr.datasets.preparers.gatherers'])
+
+# manage data parser
+DATA_PARSERS = Registry(
+    'data parser', locations=['mmocr.datasets.preparers.parsers'])
+
+# manage data packer
+DATA_PACKERS = Registry(
+    'data packer', locations=['mmocr.datasets.preparers.packers'])
+
+# manage data dumper
+DATA_DUMPERS = Registry(
+    'data dumper', locations=['mmocr.datasets.preparers.dumpers'])
+
+# manage dataset config generator
+CFG_GENERATORS = Registry(
+    'cfg generator', locations=['mmocr.datasets.preparers.config_generators'])