print("HIIIIIIOOO") import importlib from .base_model import BaseModel print("HIIIIIIOOO") def find_model_using_name(model_name): # Given the option --model [modelname], # the file "models/modelname_model.py" # will be imported. # task_module = importlib.import_module(task_name) # model_filename = task_name + ".models." + model_name.lower() + "_model" model_filename = "models." + model_name.lower() + "_model" # modellib = importlib.import_module(model_filename, package=task_module) modellib = importlib.import_module(model_filename) # In the file, the class called ModelNameModel() will # be instantiated. It has to be a subclass of BaseModel, # and it is case-insensitive. model = None target_model_name = model_name.replace('_', '') + 'model' for name, cls in modellib.__dict__.items(): if name.lower() == target_model_name.lower() \ and next(iter(cls.__bases__)).__module__.endswith(BaseModel.__module__): # check that base class is BaseModel model = cls if model is None: raise NotImplementedError("In %s.py, there should be a subclass of BaseModel with class name that matches %s in lowercase." % (model_filename, target_model_name)) return model def get_option_setter(model_name): model_class = find_model_using_name(model_name) return model_class.modify_commandline_options def create_model(opt): instance = create_model_by_name(opt.model, opt) return instance def create_model_by_name(name, opt): model = find_model_using_name(name) instance = model(opt) print("model [%s] was created" % (instance.name())) return instance