diff --git a/mmdeploy/codebase/mmcls/deploy/classification_model.py b/mmdeploy/codebase/mmcls/deploy/classification_model.py index 458c66ef8..f7f3bbfe7 100644 --- a/mmdeploy/codebase/mmcls/deploy/classification_model.py +++ b/mmdeploy/codebase/mmcls/deploy/classification_model.py @@ -98,7 +98,7 @@ class End2EndModel(BaseBackendModel): def show_result(self, img: np.ndarray, result: list, - win_name: str, + win_name: str = '', show: bool = True, out_file: str = None): """Show predictions of classification. diff --git a/mmdeploy/codebase/mmocr/deploy/mmocr.py b/mmdeploy/codebase/mmocr/deploy/mmocr.py index 5bc95bf2c..c97d2958a 100644 --- a/mmdeploy/codebase/mmocr/deploy/mmocr.py +++ b/mmdeploy/codebase/mmocr/deploy/mmocr.py @@ -4,6 +4,7 @@ from typing import Optional, Union import mmcv import torch from mmcv.utils import Registry +from packaging import version from torch.utils.data import DataLoader, Dataset from mmdeploy.codebase.base import CODEBASE, BaseTask, MMCodebase @@ -137,6 +138,11 @@ class MMOCR(MMCodebase): Returns: list: The prediction results. """ - from mmdet.apis import single_gpu_test + import mmocr + # fixed the bug when using `--show-dir` after mocr v0.4.1 + if version.parse(mmocr.__version__) < version.parse('0.4.1'): + from mmdet.apis import single_gpu_test + else: + from mmocr.apis import single_gpu_test outputs = single_gpu_test(model, data_loader, show, out_dir, **kwargs) return outputs diff --git a/mmdeploy/codebase/mmocr/deploy/text_detection_model.py b/mmdeploy/codebase/mmocr/deploy/text_detection_model.py index 4b9b4ec99..31861b66e 100644 --- a/mmdeploy/codebase/mmocr/deploy/text_detection_model.py +++ b/mmdeploy/codebase/mmocr/deploy/text_detection_model.py @@ -118,7 +118,7 @@ class End2EndModel(BaseBackendModel): def show_result(self, img: np.ndarray, result: dict, - win_name: str, + win_name: str = '', show: bool = True, score_thr: float = 0.3, out_file: str = None): diff --git a/mmdeploy/codebase/mmocr/deploy/text_recognition_model.py b/mmdeploy/codebase/mmocr/deploy/text_recognition_model.py index 42e88f223..7f07dbba6 100644 --- a/mmdeploy/codebase/mmocr/deploy/text_recognition_model.py +++ b/mmdeploy/codebase/mmocr/deploy/text_recognition_model.py @@ -125,7 +125,7 @@ class End2EndModel(BaseBackendModel): def show_result(self, img: np.ndarray, result: list, - win_name: str, + win_name: str = '', show: bool = True, score_thr: float = 0.3, out_file: str = None):