mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
ef4b30038f
* add DPT head * [fix] fix init error * use mmcv function * delete code * remove transpose clas * support NLC output shape * Delete post_process_layer.py * add unittest and docstring * rename variables * fix project error and add unittest * match dpt weights * add configs * fix vit pos_embed bug and dpt feature fusion bug * match vit output * fix gelu * minor change * update unitest * fix configs error * inference test * remove auxilary * use local pretrain * update training results * update yml * update fps and memory test * update doc * update readme * add yml * update doc * remove with_cp * update config * update docstring * remove dpt-l * add init_cfg and modify readme.md * Update dpt_vit-b16.py * zh-n README * use constructor instead of build function * prevent tensor being modified by ConvModule * fix unittest Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
35 lines
1.2 KiB
Python
35 lines
1.2 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 .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 .sep_aspp_head import DepthwiseSeparableASPPHead
|
|
from .sep_fcn_head import DepthwiseSeparableFCNHead
|
|
from .setr_mla_head import SETRMLAHead
|
|
from .setr_up_head import SETRUPHead
|
|
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', 'SegformerHead'
|
|
]
|