mmclassification/mmpretrain/apis/__init__.py

13 lines
497 B
Python

# 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'
]