mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
2e28db0dcb
* knet first commit * fix import error in knet * remove kernel update head from decoder head * [Feature] Add kenerl updation for some decoder heads. * [Feature] Add kenerl updation for some decoder heads. * directly use forward_feature && modify other 3 decoder heads * remover kernel_update attr * delete unnecessary variables in forward function * delete kernel update function * delete kernel update function * delete kernel_generate_head * add unit test & comments in knet.py * add copyright to fix lint error * modify config names of knet * rename swin-l 640 * upload models&logs and refactor knet_head.py * modify docstrings and add some ut * add url, modify docstring and add loss ut * modify docstrings
41 lines
1.5 KiB
Python
41 lines
1.5 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 .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'
|
|
]
|