[Fix] mmocr.utils.typing -> mmocr.utils.typing_utils (#1538)

pull/1546/head
Tong Gao 2022-11-17 10:21:00 +08:00 committed by GitHub
parent d8c0df4827
commit e067ddea23
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
16 changed files with 22 additions and 22 deletions

View File

@ -9,7 +9,7 @@ from mmengine.evaluator import BaseMetric, Evaluator
from mmengine.evaluator.metric import _to_cpu from mmengine.evaluator.metric import _to_cpu
from mmocr.registry import EVALUATOR from mmocr.registry import EVALUATOR
from mmocr.utils.typing import ConfigType from mmocr.utils.typing_utils import ConfigType
@EVALUATOR.register_module() @EVALUATOR.register_module()

View File

@ -6,8 +6,8 @@ import torch
from mmengine.model import BaseModel from mmengine.model import BaseModel
from torch import Tensor from torch import Tensor
from mmocr.utils.typing import (DetSampleList, OptConfigType, OptDetSampleList, from mmocr.utils.typing_utils import (DetSampleList, OptConfigType,
OptMultiConfig) OptDetSampleList, OptMultiConfig)
ForwardResults = Union[Dict[str, torch.Tensor], DetSampleList, ForwardResults = Union[Dict[str, torch.Tensor], DetSampleList,
Tuple[torch.Tensor], torch.Tensor] Tuple[torch.Tensor], torch.Tensor]

View File

@ -11,7 +11,7 @@ from mmengine.structures import InstanceData
from mmocr.registry import MODELS from mmocr.registry import MODELS
from mmocr.utils.bbox_utils import bbox2poly from mmocr.utils.bbox_utils import bbox2poly
from mmocr.utils.typing import DetSampleList from mmocr.utils.typing_utils import DetSampleList
ForwardResults = Union[Dict[str, torch.Tensor], List[DetDataSample], ForwardResults = Union[Dict[str, torch.Tensor], List[DetDataSample],
Tuple[torch.Tensor], torch.Tensor] Tuple[torch.Tensor], torch.Tensor]

View File

@ -6,7 +6,7 @@ from mmengine.model import BaseModule
from torch import Tensor from torch import Tensor
from mmocr.registry import MODELS from mmocr.registry import MODELS
from mmocr.utils.typing import DetSampleList from mmocr.utils.typing_utils import DetSampleList
@MODELS.register_module() @MODELS.register_module()

View File

@ -9,7 +9,7 @@ from torch import Tensor
from mmocr.models.textdet.heads import BaseTextDetHead from mmocr.models.textdet.heads import BaseTextDetHead
from mmocr.registry import MODELS from mmocr.registry import MODELS
from mmocr.structures import TextDetDataSample from mmocr.structures import TextDetDataSample
from mmocr.utils.typing import DetSampleList from mmocr.utils.typing_utils import DetSampleList
@MODELS.register_module() @MODELS.register_module()

View File

@ -6,7 +6,7 @@ import torch
from torch import nn from torch import nn
from mmocr.registry import MODELS from mmocr.registry import MODELS
from mmocr.utils.typing import DetSampleList from mmocr.utils.typing_utils import DetSampleList
INPUT_TYPES = Union[torch.Tensor, Sequence[torch.Tensor], Dict] INPUT_TYPES = Union[torch.Tensor, Sequence[torch.Tensor], Dict]

View File

@ -11,7 +11,7 @@ from torch import Tensor
from mmocr.registry import MODELS from mmocr.registry import MODELS
from mmocr.structures import TextDetDataSample from mmocr.structures import TextDetDataSample
from mmocr.utils import offset_polygon from mmocr.utils import offset_polygon
from mmocr.utils.typing import ArrayLike from mmocr.utils.typing_utils import ArrayLike
from .seg_based_module_loss import SegBasedModuleLoss from .seg_based_module_loss import SegBasedModuleLoss

View File

@ -10,7 +10,7 @@ from numpy.linalg import norm
from mmocr.registry import MODELS from mmocr.registry import MODELS
from mmocr.structures import TextDetDataSample from mmocr.structures import TextDetDataSample
from mmocr.utils.typing import ArrayLike from mmocr.utils.typing_utils import ArrayLike
from .textsnake_module_loss import TextSnakeModuleLoss from .textsnake_module_loss import TextSnakeModuleLoss

View File

@ -6,7 +6,7 @@ from mmdet.models.backbones import MobileNetV2 as MMDet_MobileNetV2
from torch import Tensor from torch import Tensor
from mmocr.registry import MODELS from mmocr.registry import MODELS
from mmocr.utils.typing import InitConfigType from mmocr.utils.typing_utils import InitConfigType
@MODELS.register_module() @MODELS.register_module()

View File

@ -5,9 +5,9 @@ from typing import Dict
import torch import torch
from mmocr.registry import MODELS from mmocr.registry import MODELS
from mmocr.utils.typing import (ConfigType, InitConfigType, OptConfigType, from mmocr.utils.typing_utils import (ConfigType, InitConfigType,
OptRecSampleList, RecForwardResults, OptConfigType, OptRecSampleList,
RecSampleList) RecForwardResults, RecSampleList)
from .base import BaseRecognizer from .base import BaseRecognizer

View File

@ -20,11 +20,11 @@ from .polygon_utils import (boundary_iou, crop_polygon, is_poly_inside_rect,
sort_vertex, sort_vertex8) sort_vertex, sort_vertex8)
from .setup_env import register_all_modules from .setup_env import register_all_modules
from .string_utils import StringStripper from .string_utils import StringStripper
from .typing import (ColorType, ConfigType, DetSampleList, InitConfigType, from .typing_utils import (ColorType, ConfigType, DetSampleList,
KIESampleList, MultiConfig, OptConfigType, InitConfigType, KIESampleList, MultiConfig,
OptDetSampleList, OptInitConfigType, OptKIESampleList, OptConfigType, OptDetSampleList, OptInitConfigType,
OptMultiConfig, OptRecSampleList, OptTensor, OptKIESampleList, OptMultiConfig, OptRecSampleList,
RecForwardResults, RecSampleList) OptTensor, RecForwardResults, RecSampleList)
__all__ = [ __all__ = [
'collect_env', 'is_3dlist', 'is_type_list', 'is_none_or_type', 'equal_len', 'collect_env', 'is_3dlist', 'is_type_list', 'is_none_or_type', 'equal_len',

View File

@ -7,7 +7,7 @@ from shapely.geometry import LineString, Point
from mmocr.utils.check_argument import is_2dlist, is_type_list from mmocr.utils.check_argument import is_2dlist, is_type_list
from mmocr.utils.point_utils import point_distance, points_center from mmocr.utils.point_utils import point_distance, points_center
from mmocr.utils.typing import ArrayLike from mmocr.utils.typing_utils import ArrayLike
def rescale_bbox(bbox: np.ndarray, def rescale_bbox(bbox: np.ndarray,

View File

@ -2,7 +2,7 @@
import cv2 import cv2
import numpy as np import numpy as np
from mmocr.utils.typing import ArrayLike from mmocr.utils.typing_utils import ArrayLike
def fill_hole(input_mask: ArrayLike) -> np.array: def fill_hole(input_mask: ArrayLike) -> np.array:

View File

@ -1,7 +1,7 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import numpy as np import numpy as np
from mmocr.utils.typing import ArrayLike from mmocr.utils.typing_utils import ArrayLike
def points_center(points: ArrayLike) -> np.ndarray: def points_center(points: ArrayLike) -> np.ndarray:

View File

@ -10,7 +10,7 @@ from shapely.geometry import MultiPolygon, Polygon
from mmocr.utils import bbox2poly, valid_boundary from mmocr.utils import bbox2poly, valid_boundary
from mmocr.utils.check_argument import is_2dlist from mmocr.utils.check_argument import is_2dlist
from mmocr.utils.typing import ArrayLike from mmocr.utils.typing_utils import ArrayLike
def rescale_polygon(polygon: ArrayLike, def rescale_polygon(polygon: ArrayLike,