DL-Art-School/codes/models/eval/__init__.py

12 lines
417 B
Python

from models.eval.sr_style import SrStyleTransferEvaluator
from models.eval.style import StyleTransferEvaluator
def create_evaluator(model, opt_eval, env):
type = opt_eval['type']
if type == 'style_transfer':
return StyleTransferEvaluator(model, opt_eval, env)
elif type == 'sr_stylegan':
return SrStyleTransferEvaluator(model, opt_eval, env)
else:
raise NotImplementedError()