mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
651da35c0e
* add unet * add unet * add unet * update test_unet * update test_unet * update test_unet * update test_unet * fix bugs * add init method for unet * add test of UNet init_weights method * add registry * merge upsample * fix test * Update mmseg/models/backbones/unet.py Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com> * Update mmseg/models/backbones/unet.py Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com> * split UpConvBlock from UNet * use reversed * rename upsample module * rename upsample module * rename upsample module * rename upsample module Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
11 lines
324 B
Python
11 lines
324 B
Python
from .inverted_residual import InvertedResidual
|
|
from .make_divisible import make_divisible
|
|
from .res_layer import ResLayer
|
|
from .self_attention_block import SelfAttentionBlock
|
|
from .up_conv_block import UpConvBlock
|
|
|
|
__all__ = [
|
|
'ResLayer', 'SelfAttentionBlock', 'make_divisible', 'InvertedResidual',
|
|
'UpConvBlock'
|
|
]
|