mmseg/models/backbones/__init__.py

from .cgnet import CGNet
from .fast_scnn import FastSCNN
from .hrnet import HRNet
from .mobilenet_v2 import MobileNetV2
from .mobilenet_v3 import MobileNetV3
from .resnest import ResNeSt
from .resnet import ResNet, ResNetV1c, ResNetV1d
from .resnext import ResNeXt
from .unet import UNet
from .pvt import pvt_small, pvt_small_f4, pvt_tiny
from .vit import VisionTransformer
from .vit_mla import VIT_MLA

__all__ = [
    'ResNet', 'ResNetV1c', 'ResNetV1d', 'ResNeXt', 'HRNet', 'FastSCNN',
    'ResNeSt', 'MobileNetV2', 'UNet', 'CGNet', 'MobileNetV3', 
    'pvt_small', 'pvt_small_f4', 'pvt_tiny', 
    'VisionTransformer', 'VIT_MLA'
]