mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
542143cb41
* [Feature] Add TIMM and HuggingFace wrappers to build classifiers from them directly. * Support `with_cp` and add docstring. * Add unit tests. * Update CI. * Update docs.
11 lines
299 B
Python
11 lines
299 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .base import BaseClassifier
|
|
from .hugging_face import HuggingFaceClassifier
|
|
from .image import ImageClassifier
|
|
from .timm import TimmClassifier
|
|
|
|
__all__ = [
|
|
'BaseClassifier', 'ImageClassifier', 'TimmClassifier',
|
|
'HuggingFaceClassifier'
|
|
]
|