mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
c46cc85cba
Thanks for your contribution and we appreciate it a lot. The following instructions would make your pull request more healthy and more easily get feedback. If you do not understand some items, don't worry, just make the pull request and seek help from maintainers. ## Motivation Support depth estimation algorithm [VPD](https://github.com/wl-zhao/VPD) ## Modification 1. add VPD backbone 2. add VPD decoder head for depth estimation 3. add a new segmentor `DepthEstimator` based on `EncoderDecoder` for depth estimation 4. add an integrated metric that calculate common metrics in depth estimation 5. add SiLog loss for depth estimation 6. add config for VPD ## BC-breaking (Optional) Does the modification introduce changes that break the backward-compatibility of the downstream repos? If so, please describe how it breaks the compatibility and how the downstream projects should modify their code to keep compatibility with this PR. ## Use cases (Optional) If this PR introduces a new feature, it is better to list some use cases here, and update the documentation. ## Checklist 1. Pre-commit or other linting tools are used to fix the potential lint issues. 7. The modification is covered by complete unit tests. If not, please add more unit test to ensure the correctness. 8. If the modification has potential influence on downstream projects, this PR should be tested with downstream projects, like MMDet or MMDet3D. 9. The documentation has been modified accordingly, like docstring or example tutorials.
48 lines
1.9 KiB
Python
48 lines
1.9 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 .ddr_head import DDRHead
|
|
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 .ham_head import LightHamHead
|
|
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 .pid_head import PIDHead
|
|
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
|
|
from .vpd_depth_head import VPDDepthHead
|
|
|
|
__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',
|
|
'LightHamHead', 'PIDHead', 'DDRHead', 'VPDDepthHead'
|
|
]
|