RabbitRUI/ruispace

1from .iresnet import iresnet18, iresnet34, iresnet50, iresnet100, iresnet2002from .mobilefacenet import get_mbf3 4 5def get_model(name, **kwargs):6    # resnet7    if name == "r18":8        return iresnet18(False, **kwargs)9    elif name == "r34":10        return iresnet34(False, **kwargs)11    elif name == "r50":12        return iresnet50(False, **kwargs)13    elif name == "r100":14        return iresnet100(False, **kwargs)15    elif name == "r200":16        return iresnet200(False, **kwargs)17    elif name == "r2060":18        from .iresnet2060 import iresnet206019        return iresnet2060(False, **kwargs)20    elif name == "mbf":21        fp16 = kwargs.get("fp16", False)22        num_features = kwargs.get("num_features", 512)23        return get_mbf(fp16=fp16, num_features=num_features)24    else:25        raise ValueError()