mmclassification/mmpretrain/apis/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .feature_extractor import FeatureExtractor
from .image_classification import ImageClassificationInferencer
from .image_retrieval import ImageRetrievalInferencer
from .model import (ModelHub, get_model, inference_model, init_model,
                    list_models)

__all__ = [
    'init_model', 'inference_model', 'list_models', 'get_model', 'ModelHub',
    'ImageClassificationInferencer', 'ImageRetrievalInferencer',
    'FeatureExtractor'
]