skp/models/segmentation/__init__.py

from .decoders.deeplabv3 import DeepLabV3Plus
from .decoders.deeplabv3_3d import DeepLabV3Plus_3D
from .decoders.fpn import FPN
from .decoders.nas_fpn import NASFPN
from .decoders.unet import Unet
from .decoders.unet_3d import Unet_3D