mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
414ba80274
* [Refactor] Refactor APIs, add `ImageRetrievalInferencer` and `FeatureExtractor'. * Update image retrieval * Update FeatureExtractor * Fix UT
13 lines
497 B
Python
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'
|
|
]
|