mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
9038c1c255
* [Feature] Support TTA and add `--tta` in `tools/test.py`. * Add unit tests. * Rename the TTA model to `AverageClsScoreTTA`.
18 lines
722 B
Python
18 lines
722 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .backbones import * # noqa: F401,F403
|
|
from .builder import (BACKBONES, CLASSIFIERS, HEADS, LOSSES, NECKS,
|
|
build_backbone, build_classifier, build_head, build_loss,
|
|
build_neck)
|
|
from .classifiers import * # noqa: F401,F403
|
|
from .heads import * # noqa: F401,F403
|
|
from .losses import * # noqa: F401,F403
|
|
from .necks import * # noqa: F401,F403
|
|
from .retrievers import * # noqa: F401,F403
|
|
from .tta import * # noqa: F401,F403
|
|
from .utils import * # noqa: F401,F403
|
|
|
|
__all__ = [
|
|
'BACKBONES', 'HEADS', 'NECKS', 'LOSSES', 'CLASSIFIERS', 'build_backbone',
|
|
'build_head', 'build_neck', 'build_loss', 'build_classifier'
|
|
]
|