fix: import YoloxPredictor (#205)

* import yoloxpredictor
This commit is contained in:
wenmeng zhou 2022-10-12 11:32:26 +08:00 committed by GitHub
parent 513e0d060a
commit 1e1afbfd0e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 2 deletions

View File

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

View File

@ -36,7 +36,8 @@ try:
}
}
MM_ORIGINAL_REGISTRY = copy.deepcopy(MM_REGISTRY)
except ImportError:
except ImportError as e:
print(e)
pass
EASYCV_REGISTRY_MAP = {