File size: 1,142 Bytes
ec0fdfd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
"""This package contains modules related to function, network architectures, and models"""
import importlib
from .base_model import BaseModel
def find_model_using_name(model_name):
"""Import the module "model/[model_name]_model.py"."""
model_file_name = "model." + model_name + "_model"
modellib = importlib.import_module(model_file_name)
model = None
for name, cls in modellib.__dict__.items():
if name.lower() == model_name.lower() and issubclass(cls, BaseModel):
model = cls
if model is None:
print("In %s.py, there should be a subclass of BaseModel with class name that matches %s in lowercase." % (model_file_name, model_name))
exit(0)
return model
def get_option_setter(model_name):
"""Return the static method <modify_commandline_options> of the model class."""
model = find_model_using_name(model_name)
return model.modify_options
def create_model(opt):
"""Create a model given the option."""
model = find_model_using_name(opt.model)
instance = model(opt)
print("model [%s] was created" % type(instance).__name__)
return instance |