12 lines
310 B
Python
12 lines
310 B
Python
|
from __future__ import absolute_import
|
||
|
|
from networks.AugmentCE2P import resnet101
|
||
|
|
|
||
|
|
__factory = {
|
||
|
|
'resnet101': resnet101,
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def init_model(name, *args, **kwargs):
|
||
|
|
if name not in __factory.keys():
|
||
|
|
raise KeyError("Unknown model arch: {}".format(name))
|
||
|
|
return __factory[name](*args, **kwargs)
|