diff --git a/mmdeploy/codebase/mmocr/deploy/text_recognition_model.py b/mmdeploy/codebase/mmocr/deploy/text_recognition_model.py index 7fe006d15..7b846ce18 100644 --- a/mmdeploy/codebase/mmocr/deploy/text_recognition_model.py +++ b/mmdeploy/codebase/mmocr/deploy/text_recognition_model.py @@ -5,7 +5,7 @@ import mmengine import torch from mmengine.registry import Registry from mmengine.structures import LabelData -from mmocr.utils.typing import RecSampleList +from mmocr.utils.typing_utils import RecSampleList from mmdeploy.codebase.base import BaseBackendModel from mmdeploy.utils import (Backend, get_backend, get_codebase_config, diff --git a/mmdeploy/codebase/mmocr/models/text_detection/heads.py b/mmdeploy/codebase/mmocr/models/text_detection/heads.py index b7d855f90..9c5801852 100644 --- a/mmdeploy/codebase/mmocr/models/text_detection/heads.py +++ b/mmdeploy/codebase/mmocr/models/text_detection/heads.py @@ -2,7 +2,7 @@ from typing import Dict import torch -from mmocr.utils.typing import DetSampleList +from mmocr.utils.typing_utils import DetSampleList from mmdeploy.core import FUNCTION_REWRITER diff --git a/mmdeploy/codebase/mmocr/models/text_recognition/sar_decoder.py b/mmdeploy/codebase/mmocr/models/text_recognition/sar_decoder.py index bc371d81c..c6fc68c83 100644 --- a/mmdeploy/codebase/mmocr/models/text_recognition/sar_decoder.py +++ b/mmdeploy/codebase/mmocr/models/text_recognition/sar_decoder.py @@ -4,7 +4,7 @@ from typing import Optional, Sequence import torch import torch.nn.functional as F -from mmocr.utils.typing import TextRecogDataSample +from mmocr.utils.typing_utils import TextRecogDataSample from torch import nn from mmdeploy.core import FUNCTION_REWRITER, MODULE_REWRITER