def getOpt(optDe):
model = optDe.get('model', 'dehaze')
opt = Option()
modelPath, opt.modelDef, opt.ram, opt.padding, opt.align, opt.prepare = mode_switch[
model]
opt.model = modelPath
opt.modelCached = initModel(opt, modelPath, model)
return opt
def getOpt(optSR):
opt = Option()
opt.mode = optSR['model']
opt.scale = optSR['scale']
nmode = opt.mode + str(opt.scale)
if not nmode in mode_switch:
return
if opt.mode[:3] != 'gan':
opt.squeeze = lambda x: x.squeeze(1)
opt.unsqueeze = lambda x: x.unsqueeze(1)
opt.padding = 9 if opt.scale == 3 else 5
opt.model = mode_switch[nmode][0]
opt.modelDef = mode_switch[nmode][1]
opt.ensemble = optSR['ensemble'] if 'ensemble' in optSR and (
0 <= optSR['ensemble'] <= 7) else config.ensembleSR
opt.ramCoef = mode_switch[nmode][2][config.getRunType()]
opt.cropsize = config.getConfig()[0]
opt.modelCached = initModel(opt, opt.model, 'SR' + nmode)
return opt