diff --git a/easycv/predictors/__init__.py b/easycv/predictors/__init__.py index d813dc73..5aadd60d 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,