refactor loop content

pull/1178/head
liukuikun 2022-07-12 13:43:31 +08:00 committed by gaotongxiao
parent 2d478ea244
commit 5381b1d105
6 changed files with 4 additions and 2 deletions

View File

@ -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,

View File

@ -0,0 +1,2 @@
# Copyright (c) OpenMMLab. All rights reserved.
from .runner import * # NOQA

View File

@ -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

View File

@ -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()