mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 14:59:18 +08:00
9506241f73
* feat: add arcface head * feat: add arcface head * update arcface * refactor archface head * update archface head * update archface head * use torch.cat instead of torch.hstack to fix ci Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
18 lines
744 B
Python
18 lines
744 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .arcface_head import ArcFaceClsHead
|
|
from .cls_head import ClsHead
|
|
from .conformer_head import ConformerHead
|
|
from .deit_head import DeiTClsHead
|
|
from .efficientformer_head import EfficientFormerClsHead
|
|
from .linear_head import LinearClsHead
|
|
from .multi_label_cls_head import MultiLabelClsHead
|
|
from .multi_label_linear_head import MultiLabelLinearClsHead
|
|
from .stacked_head import StackedLinearClsHead
|
|
from .vision_transformer_head import VisionTransformerClsHead
|
|
|
|
__all__ = [
|
|
'ClsHead', 'LinearClsHead', 'StackedLinearClsHead', 'MultiLabelClsHead',
|
|
'MultiLabelLinearClsHead', 'VisionTransformerClsHead', 'DeiTClsHead',
|
|
'ConformerHead', 'EfficientFormerClsHead', 'ArcFaceClsHead'
|
|
]
|