mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 21:54:47 +08:00
Rename TextRecog directories
This commit is contained in:
parent
f820a50752
commit
be30df5d50
@ -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
|
||||||
|
@ -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
|
||||||
|
|
||||||
|
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
@ -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__)
|
||||||
|
@ -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
|
||||||
|
@ -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)
|
||||||
|
|
||||||
|
|
||||||
|
@ -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)
|
||||||
|
|
||||||
|
|
||||||
|
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
@ -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):
|
||||||
|
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user