mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
163277bfe0
* init commits * fix crop size * add seg_data2instance_data method * add ut and update requirement * update configs and readme * add model-indel * update optional requirements * fix results * fix lint error * update results * update results * remove mmdet from requirements/optional.txt * use try import and update README * add docstring to overwrtied method * minor change Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .ann_head import ANNHead
|
|
from .apc_head import APCHead
|
|
from .aspp_head import ASPPHead
|
|
from .cc_head import CCHead
|
|
from .da_head import DAHead
|
|
from .dm_head import DMHead
|
|
from .dnl_head import DNLHead
|
|
from .dpt_head import DPTHead
|
|
from .ema_head import EMAHead
|
|
from .enc_head import EncHead
|
|
from .fcn_head import FCNHead
|
|
from .fpn_head import FPNHead
|
|
from .gc_head import GCHead
|
|
from .isa_head import ISAHead
|
|
from .knet_head import IterativeDecodeHead, KernelUpdateHead, KernelUpdator
|
|
from .lraspp_head import LRASPPHead
|
|
from .mask2former_head import Mask2FormerHead
|
|
from .maskformer_head import MaskFormerHead
|
|
from .nl_head import NLHead
|
|
from .ocr_head import OCRHead
|
|
from .point_head import PointHead
|
|
from .psa_head import PSAHead
|
|
from .psp_head import PSPHead
|
|
from .segformer_head import SegformerHead
|
|
from .segmenter_mask_head import SegmenterMaskTransformerHead
|
|
from .sep_aspp_head import DepthwiseSeparableASPPHead
|
|
from .sep_fcn_head import DepthwiseSeparableFCNHead
|
|
from .setr_mla_head import SETRMLAHead
|
|
from .setr_up_head import SETRUPHead
|
|
from .stdc_head import STDCHead
|
|
from .uper_head import UPerHead
|
|
|
|
__all__ = [
|
|
'FCNHead', 'PSPHead', 'ASPPHead', 'PSAHead', 'NLHead', 'GCHead', 'CCHead',
|
|
'UPerHead', 'DepthwiseSeparableASPPHead', 'ANNHead', 'DAHead', 'OCRHead',
|
|
'EncHead', 'DepthwiseSeparableFCNHead', 'FPNHead', 'EMAHead', 'DNLHead',
|
|
'PointHead', 'APCHead', 'DMHead', 'LRASPPHead', 'SETRUPHead',
|
|
'SETRMLAHead', 'DPTHead', 'SETRMLAHead', 'SegmenterMaskTransformerHead',
|
|
'SegformerHead', 'ISAHead', 'STDCHead', 'IterativeDecodeHead',
|
|
'KernelUpdateHead', 'KernelUpdator', 'MaskFormerHead', 'Mask2FormerHead'
|
|
]
|