Rename TextRecog directories

This commit is contained in:
gaotongxiao 2022-05-20 06:47:54 +00:00
parent f820a50752
commit be30df5d50
28 changed files with 20 additions and 21 deletions

View File

@ -3,17 +3,17 @@
# For example: # For example:
# mmocr/models/textdet/postprocess/utils.py # mmocr/models/textdet/postprocess/utils.py
# .*/utils.py # .*/utils.py
mmocr/models/textrecog/recognizer/base.py mmocr/models/textrecog/recognizers/base.py
.*/__init__.py .*/__init__.py
# It will be removed after all transforms have been refactored into processing.py # It will be removed after all transforms have been refactored into processing.py
mmocr/datasets/pipelines/transforms.py mmocr/datasets/pipelines/transforms.py
mmocr/datasets/pipelines/dbnet_transforms.py mmocr/datasets/pipelines/dbnet_transforms.py
# will be deleted # will be deleted
mmocr/models/textdet/dense_heads/head_mixin.py mmocr/models/textdet/heads/head_mixin.py
# Will be covered by det head tests # Will be covered by det head tests
mmocr/models/textdet/dense_heads/base_textdet_head.py mmocr/models/textdet/heads/base_textdet_head.py
# They will be removed later all det models have been refactored # They will be removed later all det models have been refactored
mmocr/models/common/detectors/single_stage.py mmocr/models/common/detectors/single_stage.py

View File

@ -10,7 +10,7 @@ from mmocr.models.textdet.detectors.single_stage_text_detector import \
SingleStageTextDetector SingleStageTextDetector
from mmocr.models.textdet.detectors.text_detector_mixin import \ from mmocr.models.textdet.detectors.text_detector_mixin import \
TextDetectorMixin TextDetectorMixin
from mmocr.models.textrecog.recognizer.encode_decode_recognizer import \ from mmocr.models.textrecog.recognizers.encode_decode_recognizer import \
EncodeDecodeRecognizer EncodeDecodeRecognizer
from mmocr.registry import MODELS from mmocr.registry import MODELS

View File

@ -1,5 +1,5 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from mmocr.models.textrecog.recognizer.base import BaseRecognizer from mmocr.models.textrecog.recognizers.base import BaseRecognizer
from mmocr.registry import MODELS from mmocr.registry import MODELS

View File

@ -1,7 +1,7 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from . import (backbones, convertors, decoders, dictionary, encoders, fusers, from . import (backbones, convertors, decoders, dictionary, encoders, fusers,
heads, losses, necks, plugins, postprocessor, preprocessor, heads, losses, necks, plugins, postprocessors, preprocessors,
recognizer) recognizers)
from .backbones import * # NOQA from .backbones import * # NOQA
from .convertors import * # NOQA from .convertors import * # NOQA
from .decoders import * # NOQA from .decoders import * # NOQA
@ -12,12 +12,12 @@ from .heads import * # NOQA
from .losses import * # NOQA from .losses import * # NOQA
from .necks import * # NOQA from .necks import * # NOQA
from .plugins import * # NOQA from .plugins import * # NOQA
from .postprocessor import * # NOQA from .postprocessors import * # NOQA
from .preprocessor import * # NOQA from .preprocessors import * # NOQA
from .recognizer import * # NOQA from .recognizers import * # NOQA
__all__ = ( __all__ = (
backbones.__all__ + convertors.__all__ + decoders.__all__ + backbones.__all__ + convertors.__all__ + decoders.__all__ +
encoders.__all__ + heads.__all__ + losses.__all__ + necks.__all__ + encoders.__all__ + heads.__all__ + losses.__all__ + necks.__all__ +
preprocessor.__all__ + recognizer.__all__ + fusers.__all__ + preprocessors.__all__ + recognizers.__all__ + fusers.__all__ +
plugins.__all__ + dictionary.__all__ + postprocessor.__all__) plugins.__all__ + dictionary.__all__ + postprocessors.__all__)

View File

@ -25,7 +25,7 @@ from mmocr.core.visualize import det_recog_show_result
from mmocr.datasets.kie_dataset import KIEDataset from mmocr.datasets.kie_dataset import KIEDataset
from mmocr.datasets.pipelines.crop import crop_img from mmocr.datasets.pipelines.crop import crop_img
from mmocr.models.textdet.detectors import TextDetectorMixin from mmocr.models.textdet.detectors import TextDetectorMixin
from mmocr.models.textrecog.recognizer import BaseRecognizer from mmocr.models.textrecog.recognizers import BaseRecognizer
from mmocr.registry import MODELS from mmocr.registry import MODELS
from mmocr.utils import is_type_list from mmocr.utils import is_type_list
from mmocr.utils.box_util import stitch_boxes_into_lines from mmocr.utils.box_util import stitch_boxes_into_lines

View File

@ -2,7 +2,7 @@
import pytest import pytest
import torch import torch
from mmocr.models.textrecog.preprocessor import (BasePreprocessor, from mmocr.models.textrecog.preprocessors import (BasePreprocessor,
TPSPreprocessor) TPSPreprocessor)

View File

@ -8,7 +8,7 @@ import pytest
import torch import torch
from mmdet.core import BitmapMasks from mmdet.core import BitmapMasks
from mmocr.models.textrecog.recognizer import (EncodeDecodeRecognizer, from mmocr.models.textrecog.recognizers import (EncodeDecodeRecognizer,
SegRecognizer) SegRecognizer)

View File

@ -9,7 +9,7 @@ from mmengine.data import LabelData
from mmocr.core.data_structures import TextRecogDataSample from mmocr.core.data_structures import TextRecogDataSample
from mmocr.models.textrecog.dictionary import Dictionary from mmocr.models.textrecog.dictionary import Dictionary
from mmocr.models.textrecog.postprocessor.attn_postprocessor import \ from mmocr.models.textrecog.postprocessors.attn_postprocessor import \
AttentionPostprocessor AttentionPostprocessor

View File

@ -8,8 +8,7 @@ from mmengine.data import LabelData
from mmocr.core.data_structures import TextRecogDataSample from mmocr.core.data_structures import TextRecogDataSample
from mmocr.models.textrecog.dictionary import Dictionary from mmocr.models.textrecog.dictionary import Dictionary
from mmocr.models.textrecog.postprocessor.base_textrecog_postprocessor import \ from mmocr.models.textrecog.postprocessors import BaseTextRecogPostprocessor
BaseTextRecogPostprocessor
class TestBaseTextRecogPostprocessor(TestCase): class TestBaseTextRecogPostprocessor(TestCase):

View File

@ -9,7 +9,7 @@ from mmengine.data import LabelData
from mmocr.core.data_structures import TextRecogDataSample from mmocr.core.data_structures import TextRecogDataSample
from mmocr.models.textrecog.dictionary import Dictionary from mmocr.models.textrecog.dictionary import Dictionary
from mmocr.models.textrecog.postprocessor.ctc_postprocessor import \ from mmocr.models.textrecog.postprocessors.ctc_postprocessor import \
CTCPostProcessor CTCPostProcessor