8ff866d60c
* Add DNLHead * add configs * add weight decay mult * add norm back * Update README.md * matched inference performance * Fixed shape * sep conv_out * no norm * add norm back * complete model zoo * add tests * Add test forward * Add more test Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
24 lines
800 B
Python
24 lines
800 B
Python
from .ann_head import ANNHead
|
|
from .aspp_head import ASPPHead
|
|
from .cc_head import CCHead
|
|
from .da_head import DAHead
|
|
from .dnl_head import DNLHead
|
|
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 .nl_head import NLHead
|
|
from .ocr_head import OCRHead
|
|
from .psa_head import PSAHead
|
|
from .psp_head import PSPHead
|
|
from .sep_aspp_head import DepthwiseSeparableASPPHead
|
|
from .sep_fcn_head import DepthwiseSeparableFCNHead
|
|
from .uper_head import UPerHead
|
|
|
|
__all__ = [
|
|
'FCNHead', 'PSPHead', 'ASPPHead', 'PSAHead', 'NLHead', 'GCHead', 'CCHead',
|
|
'UPerHead', 'DepthwiseSeparableASPPHead', 'ANNHead', 'DAHead', 'OCRHead',
|
|
'EncHead', 'DepthwiseSeparableFCNHead', 'FPNHead', 'EMAHead', 'DNLHead'
|
|
]
|