[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 mmocr.registry import EVALUATOR
from mmocr.utils.typing import ConfigType
from mmocr.utils.typing_utils import ConfigType
@EVALUATOR.register_module()

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

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

View File

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

View File

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

View File

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