mirror of
https://github.com/deepinsight/insightface.git
synced 2026-05-18 14:55:42 +00:00
20 lines
600 B
Python
20 lines
600 B
Python
from .ir_resnet import iresnet18, iresnet34, iresnet50, iresnet100, iresnet200
|
|
from .fmobilefacenet import mobilefacenet
|
|
|
|
|
|
def get_model(name, input_blob, cfg):
|
|
if name == "r18":
|
|
return iresnet18(input_blob, cfg)
|
|
elif name == "r34":
|
|
return iresnet34(input_blob, cfg)
|
|
elif name == "r50":
|
|
return iresnet50(input_blob, cfg)
|
|
elif name == "r100":
|
|
return iresnet100(input_blob, cfg)
|
|
elif name == "r200":
|
|
return iresnet200(input_blob, cfg)
|
|
elif name == "mbf":
|
|
return mobilefacenet(input_blob, cfg)
|
|
else:
|
|
raise ValueError()
|