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>
13 lines
349 B
Python
13 lines
349 B
Python
from .fast_scnn import FastSCNN
|
|
from .hrnet import HRNet
|
|
from .mobilenet_v2 import MobileNetV2
|
|
from .resnest import ResNeSt
|
|
from .resnet import ResNet, ResNetV1c, ResNetV1d
|
|
from .resnext import ResNeXt
|
|
from .unet import UNet
|
|
|
|
__all__ = [
|
|
'ResNet', 'ResNetV1c', 'ResNetV1d', 'ResNeXt', 'HRNet', 'FastSCNN',
|
|
'ResNeSt', 'MobileNetV2', 'UNet'
|
|
]
|