From 1e1afbfd0e219d4fb7061ff5e9c621250ffe5271 Mon Sep 17 00:00:00 2001 From: wenmeng zhou Date: Wed, 12 Oct 2022 11:32:26 +0800 Subject: [PATCH] fix: import YoloxPredictor (#205) * import yoloxpredictor --- easycv/predictors/__init__.py | 3 ++- easycv/utils/mmlab_utils.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/easycv/predictors/__init__.py b/easycv/predictors/__init__.py index 2d38f2f2..46c4d4df 100644 --- a/easycv/predictors/__init__.py +++ b/easycv/predictors/__init__.py @@ -1,7 +1,8 @@ # Copyright (c) Alibaba, Inc. and its affiliates. from .classifier import TorchClassifier from .detector import (DetectionPredictor, TorchFaceDetector, - TorchYoloXClassifierPredictor, TorchYoloXPredictor) + TorchYoloXClassifierPredictor, TorchYoloXPredictor, + YoloXPredictor) from .face_keypoints_predictor import FaceKeypointsPredictor from .feature_extractor import (TorchFaceAttrExtractor, TorchFaceFeatureExtractor, diff --git a/easycv/utils/mmlab_utils.py b/easycv/utils/mmlab_utils.py index e4e2df86..ca925cd5 100644 --- a/easycv/utils/mmlab_utils.py +++ b/easycv/utils/mmlab_utils.py @@ -36,7 +36,8 @@ try: } } MM_ORIGINAL_REGISTRY = copy.deepcopy(MM_REGISTRY) -except ImportError: +except ImportError as e: + print(e) pass EASYCV_REGISTRY_MAP = {