50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
import importlib
|
|
|
|
from app.service.image2sketch.models import unpaired_model as modellib
|
|
from .base_model import BaseModel
|
|
|
|
|
|
def find_model_using_name(model_name):
|
|
"""Import the module "models/[model_name]_model.py".
|
|
|
|
In the file, the class called DatasetNameModel() will
|
|
be instantiated. It has to be a subclass of BaseModel,
|
|
and it is case-insensitive.
|
|
"""
|
|
# model_filename = "." + model_name + "_model"
|
|
# modellib = importlib.import_module(model_filename)
|
|
model = None
|
|
target_model_name = model_name.replace('_', '') + 'model'
|
|
for name, cls in modellib.__dict__.items():
|
|
if name.lower() == target_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_filename, target_model_name))
|
|
exit(0)
|
|
|
|
return model
|
|
|
|
|
|
def get_option_setter(model_name):
|
|
"""Return the static method <modify_commandline_options> of the model class."""
|
|
model_class = find_model_using_name(model_name)
|
|
return model_class.modify_commandline_options
|
|
|
|
|
|
def create_model(opt):
|
|
"""Create a model given the option.
|
|
|
|
This function warps the class CustomDatasetDataLoader.
|
|
This is the main interface between this package and 'train.py'/'test.py'
|
|
|
|
Example:
|
|
>>> from .models import create_model
|
|
>>> model = create_model(opt)
|
|
"""
|
|
model = find_model_using_name(opt.model)
|
|
instance = model(opt)
|
|
print("model [%s] was created" % type(instance).__name__)
|
|
return instance
|