mirror of https://github.com/open-mmlab/mmocr.git
refactor loop content
parent
2d478ea244
commit
5381b1d105
|
@ -1,6 +1,5 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from .data_structures import * # NOQA
|
||||
from .loops import * # NOQA
|
||||
from .visualization import TextDetLocalVisualizer, TextRecogLocalVisualizer
|
||||
from .visualize import (det_recog_show_result, imshow_edge, imshow_node,
|
||||
imshow_pred_boundary, imshow_text_char_boundary,
|
||||
|
|
|
@ -0,0 +1,2 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from .runner import * # NOQA
|
|
@ -18,6 +18,7 @@ def register_all_modules(init_default_scope: bool = True) -> None:
|
|||
""" # noqa
|
||||
import mmocr.core # noqa: F401,F403
|
||||
import mmocr.datasets # noqa: F401,F403
|
||||
import mmocr.engine # noqa: F401,F403
|
||||
import mmocr.evaluation # noqa: F401,F403
|
||||
import mmocr.models # noqa: F401,F403
|
||||
|
||||
|
|
|
@ -14,7 +14,7 @@ from mmengine.registry import DATASETS, HOOKS, METRICS, MODELS
|
|||
from mmengine.runner import Runner
|
||||
from torch.utils.data import Dataset
|
||||
|
||||
from mmocr.core.loops import MultiTestLoop, MultiValLoop
|
||||
from mmocr.engine.runner import MultiTestLoop, MultiValLoop
|
||||
|
||||
|
||||
@MODELS.register_module()
|
Loading…
Reference in New Issue