<reponame>ViTAE-Transformer/ViTAE-Transformer-Matting
"""
Rethinking Portrait Matting with Privacy Preserving

Copyright (c) 2022, <NAME> (<EMAIL>) and <NAME> (<EMAIL>)
Licensed under the MIT License (see LICENSE for details)
Github repo: https://github.com/ViTAE-Transformer/ViTAE-Transformer-Matting.git
Paper link: https://arxiv.org/abs/2203.16828

"""

from .RestNet34 import *
from .Swin_T import *
from .ViTAE_S import *


def build_model(model_arch, **kwargs):
    if model_arch == "r34":
        model = p3mnet_resnet34(**kwargs)
    elif model_arch == "swin":
        model = p3mnet_swin_t(**kwargs)
    elif model_arch == "vitae":
        model = p3mnet_vitae_s(**kwargs)
    else:
        raise NotImplementedError
    
    return model
"""
Rethinking Portrait Matting with Privacy Preserving

Copyright (c) 2022, <NAME> (<EMAIL>) and <NAME> (<EMAIL>)
Licensed under the MIT License (see LICENSE for details)
Github repo: https://github.com/ViTAE-Transformer/ViTAE-Transformer-Matting.git
Paper link: https://arxiv.org/abs/2203.16828

"""

from .RestNet34 import *
from .Swin_T import *
from .ViTAE_S import *


def build_model(model_arch, **kwargs):
    if model_arch == "r34":
        model = p3mnet_resnet34(**kwargs)
    elif model_arch == "swin":
        model = p3mnet_swin_t(**kwargs)
    elif model_arch == "vitae":
        model = p3mnet_vitae_s(**kwargs)
    else:
        raise NotImplementedError
    
    return model
Rethinking Portrait Matting with Privacy Preserving  Copyright (c) 2022, <NAME> (<EMAIL>) and <NAME> (<EMAIL>) Licensed under the MIT License (see LICENSE for details) Github repo: https://github.com/ViTAE-Transformer/ViTAE-Transformer-Matting.git Paper link: https://arxiv.org/abs/2203.16828