mirror of https://github.com/open-mmlab/mmocr.git
[Fix] mmocr.utils.typing -> mmocr.utils.typing_utils (#1538)
parent
d8c0df4827
commit
e067ddea23
|
@ -9,7 +9,7 @@ from mmengine.evaluator import BaseMetric, Evaluator
|
|||
from mmengine.evaluator.metric import _to_cpu
|
||||
|
||||
from mmocr.registry import EVALUATOR
|
||||
from mmocr.utils.typing import ConfigType
|
||||
from mmocr.utils.typing_utils import ConfigType
|
||||
|
||||
|
||||
@EVALUATOR.register_module()
|
||||
|
|
|
@ -6,8 +6,8 @@ import torch
|
|||
from mmengine.model import BaseModel
|
||||
from torch import Tensor
|
||||
|
||||
from mmocr.utils.typing import (DetSampleList, OptConfigType, OptDetSampleList,
|
||||
OptMultiConfig)
|
||||
from mmocr.utils.typing_utils import (DetSampleList, OptConfigType,
|
||||
OptDetSampleList, OptMultiConfig)
|
||||
|
||||
ForwardResults = Union[Dict[str, torch.Tensor], DetSampleList,
|
||||
Tuple[torch.Tensor], torch.Tensor]
|
||||
|
|
|
@ -11,7 +11,7 @@ from mmengine.structures import InstanceData
|
|||
|
||||
from mmocr.registry import MODELS
|
||||
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],
|
||||
Tuple[torch.Tensor], torch.Tensor]
|
||||
|
|
|
@ -6,7 +6,7 @@ from mmengine.model import BaseModule
|
|||
from torch import Tensor
|
||||
|
||||
from mmocr.registry import MODELS
|
||||
from mmocr.utils.typing import DetSampleList
|
||||
from mmocr.utils.typing_utils import DetSampleList
|
||||
|
||||
|
||||
@MODELS.register_module()
|
||||
|
|
|
@ -9,7 +9,7 @@ from torch import Tensor
|
|||
from mmocr.models.textdet.heads import BaseTextDetHead
|
||||
from mmocr.registry import MODELS
|
||||
from mmocr.structures import TextDetDataSample
|
||||
from mmocr.utils.typing import DetSampleList
|
||||
from mmocr.utils.typing_utils import DetSampleList
|
||||
|
||||
|
||||
@MODELS.register_module()
|
||||
|
|
|
@ -6,7 +6,7 @@ import torch
|
|||
from torch import nn
|
||||
|
||||
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]
|
||||
|
||||
|
|
|
@ -11,7 +11,7 @@ from torch import Tensor
|
|||
from mmocr.registry import MODELS
|
||||
from mmocr.structures import TextDetDataSample
|
||||
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
|
||||
|
||||
|
||||
|
|
|
@ -10,7 +10,7 @@ from numpy.linalg import norm
|
|||
|
||||
from mmocr.registry import MODELS
|
||||
from mmocr.structures import TextDetDataSample
|
||||
from mmocr.utils.typing import ArrayLike
|
||||
from mmocr.utils.typing_utils import ArrayLike
|
||||
from .textsnake_module_loss import TextSnakeModuleLoss
|
||||
|
||||
|
||||
|
|
|
@ -6,7 +6,7 @@ from mmdet.models.backbones import MobileNetV2 as MMDet_MobileNetV2
|
|||
from torch import Tensor
|
||||
|
||||
from mmocr.registry import MODELS
|
||||
from mmocr.utils.typing import InitConfigType
|
||||
from mmocr.utils.typing_utils import InitConfigType
|
||||
|
||||
|
||||
@MODELS.register_module()
|
||||
|
|
|
@ -5,9 +5,9 @@ from typing import Dict
|
|||
import torch
|
||||
|
||||
from mmocr.registry import MODELS
|
||||
from mmocr.utils.typing import (ConfigType, InitConfigType, OptConfigType,
|
||||
OptRecSampleList, RecForwardResults,
|
||||
RecSampleList)
|
||||
from mmocr.utils.typing_utils import (ConfigType, InitConfigType,
|
||||
OptConfigType, OptRecSampleList,
|
||||
RecForwardResults, RecSampleList)
|
||||
from .base import BaseRecognizer
|
||||
|
||||
|
||||
|
|
|
@ -20,11 +20,11 @@ from .polygon_utils import (boundary_iou, crop_polygon, is_poly_inside_rect,
|
|||
sort_vertex, sort_vertex8)
|
||||
from .setup_env import register_all_modules
|
||||
from .string_utils import StringStripper
|
||||
from .typing import (ColorType, ConfigType, DetSampleList, InitConfigType,
|
||||
KIESampleList, MultiConfig, OptConfigType,
|
||||
OptDetSampleList, OptInitConfigType, OptKIESampleList,
|
||||
OptMultiConfig, OptRecSampleList, OptTensor,
|
||||
RecForwardResults, RecSampleList)
|
||||
from .typing_utils import (ColorType, ConfigType, DetSampleList,
|
||||
InitConfigType, KIESampleList, MultiConfig,
|
||||
OptConfigType, OptDetSampleList, OptInitConfigType,
|
||||
OptKIESampleList, OptMultiConfig, OptRecSampleList,
|
||||
OptTensor, RecForwardResults, RecSampleList)
|
||||
|
||||
__all__ = [
|
||||
'collect_env', 'is_3dlist', 'is_type_list', 'is_none_or_type', 'equal_len',
|
||||
|
|
|
@ -7,7 +7,7 @@ from shapely.geometry import LineString, Point
|
|||
|
||||
from mmocr.utils.check_argument import is_2dlist, is_type_list
|
||||
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,
|
||||
|
|
|
@ -2,7 +2,7 @@
|
|||
import cv2
|
||||
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:
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import numpy as np
|
||||
|
||||
from mmocr.utils.typing import ArrayLike
|
||||
from mmocr.utils.typing_utils import ArrayLike
|
||||
|
||||
|
||||
def points_center(points: ArrayLike) -> np.ndarray:
|
||||
|
|
|
@ -10,7 +10,7 @@ from shapely.geometry import MultiPolygon, Polygon
|
|||
|
||||
from mmocr.utils import bbox2poly, valid_boundary
|
||||
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,
|
||||
|
|
Loading…
Reference in New Issue