EasyCV/easycv/predictors/__init__.py
yhq 8379127388
add video recognition (#256)
* add x3d swin-video clip-bert video algorithm
2023-01-10 17:33:35 +08:00

19 lines
1.0 KiB
Python

# Copyright (c) Alibaba, Inc. and its affiliates.
from .bevformer_predictor import BEVFormerPredictor
from .classifier import TorchClassifier
from .detector import (DetectionPredictor, TorchFaceDetector,
TorchYoloXClassifierPredictor, TorchYoloXPredictor,
YoloXPredictor)
from .face_keypoints_predictor import FaceKeypointsPredictor
from .feature_extractor import (TorchFaceAttrExtractor,
TorchFaceFeatureExtractor,
TorchFeatureExtractor)
from .hand_keypoints_predictor import HandKeypointsPredictor
from .ocr import (OCRClsPredictor, OCRDetPredictor, OCRPredictor,
OCRRecPredictor)
from .pose_predictor import (TorchPoseTopDownPredictor,
TorchPoseTopDownPredictorWithDetector)
from .segmentation import Mask2formerPredictor, SegmentationPredictor
from .video_classifier import VideoClassificationPredictor
from .wholebody_keypoints_predictor import WholeBodyKeypointsPredictor