2021-03-19 12:36:15 +08:00
|
|
|
from .iresnet import iresnet18, iresnet34, iresnet50, iresnet100, iresnet200
|
2021-05-09 13:28:42 +08:00
|
|
|
|
2021-05-12 15:02:40 +08:00
|
|
|
|
2021-05-09 13:28:42 +08:00
|
|
|
def get_model(name, **kwargs):
|
|
|
|
|
if name == "r18":
|
|
|
|
|
return iresnet18(False, **kwargs)
|
|
|
|
|
elif name == "r34":
|
|
|
|
|
return iresnet34(False, **kwargs)
|
|
|
|
|
elif name == "r50":
|
|
|
|
|
return iresnet50(False, **kwargs)
|
|
|
|
|
elif name == "r100":
|
|
|
|
|
return iresnet100(False, **kwargs)
|
2021-05-12 15:02:40 +08:00
|
|
|
elif name == "r200":
|
|
|
|
|
return iresnet200(False, **kwargs)
|
2021-05-09 13:28:42 +08:00
|
|
|
else:
|
|
|
|
|
raise ValueError()
|
