Spaces:
Runtime error
Runtime error
File size: 2,110 Bytes
3b96cb1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
# Copyright (c) OpenMMLab. All rights reserved.
from .beitv1_head import BEiTV1Head
from .beitv2_head import BEiTV2Head
from .cae_head import CAEHead
from .cls_head import ClsHead
from .conformer_head import ConformerHead
from .contrastive_head import ContrastiveHead
from .deit_head import DeiTClsHead
from .efficientformer_head import EfficientFormerClsHead
from .grounding_head import GroundingHead
from .itc_head import ITCHead
from .itm_head import ITMHead
from .itpn_clip_head import iTPNClipHead
from .latent_heads import LatentCrossCorrelationHead, LatentPredictHead
from .levit_head import LeViTClsHead
from .linear_head import LinearClsHead
from .mae_head import MAEPretrainHead
from .margin_head import ArcFaceClsHead
from .mim_head import MIMHead
from .mixmim_head import MixMIMPretrainHead
from .mocov3_head import MoCoV3Head
from .multi_label_cls_head import MultiLabelClsHead
from .multi_label_csra_head import CSRAClsHead
from .multi_label_linear_head import MultiLabelLinearClsHead
from .multi_task_head import MultiTaskHead
from .seq_gen_head import SeqGenerationHead
from .simmim_head import SimMIMHead
from .spark_head import SparKPretrainHead
from .stacked_head import StackedLinearClsHead
from .swav_head import SwAVHead
from .vig_head import VigClsHead
from .vision_transformer_head import VisionTransformerClsHead
from .vqa_head import VQAGenerationHead
__all__ = [
'ClsHead',
'LinearClsHead',
'StackedLinearClsHead',
'MultiLabelClsHead',
'MultiLabelLinearClsHead',
'VisionTransformerClsHead',
'DeiTClsHead',
'ConformerHead',
'EfficientFormerClsHead',
'ArcFaceClsHead',
'CSRAClsHead',
'MultiTaskHead',
'LeViTClsHead',
'VigClsHead',
'BEiTV1Head',
'BEiTV2Head',
'CAEHead',
'ContrastiveHead',
'LatentCrossCorrelationHead',
'LatentPredictHead',
'MAEPretrainHead',
'MixMIMPretrainHead',
'SwAVHead',
'MoCoV3Head',
'MIMHead',
'SimMIMHead',
'SeqGenerationHead',
'VQAGenerationHead',
'ITCHead',
'ITMHead',
'GroundingHead',
'iTPNClipHead',
'SparKPretrainHead',
]
|