Spaces:
Running
Running
File size: 905 Bytes
2514fb4 |
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 |
"""
# --------------------------------------------
# define training model
# --------------------------------------------
"""
def define_Model(opt):
model = opt['model'] # one input: L
if model == 'plain':
from models.model_plain import ModelPlain as M
elif model == 'plain2': # two inputs: L, C
from models.model_plain2 import ModelPlain2 as M
elif model == 'plain4': # four inputs: L, k, sf, sigma
from models.model_plain4 import ModelPlain4 as M
elif model == 'gan': # one input: L
from models.model_gan import ModelGAN as M
elif model == 'vrt': # one video input L, for VRT
from models.model_vrt import ModelVRT as M
else:
raise NotImplementedError('Model [{:s}] is not defined.'.format(model))
m = M(opt)
print('Training model [{:s}] is created.'.format(m.__class__.__name__))
return m
|