Fix a bug that the alpha channel is upscaled by noise_scale model
This commit is contained in:
parent
651e8f8ecb
commit
be67a4e001
|
@ -86,7 +86,7 @@ local function convert_image(opt)
|
||||||
local model_path = path.join(opt.model_dir, ("noise%d_scale%.1fx_model.t7"):format(opt.noise_level, opt.scale))
|
local model_path = path.join(opt.model_dir, ("noise%d_scale%.1fx_model.t7"):format(opt.noise_level, opt.scale))
|
||||||
if path.exists(model_path) then
|
if path.exists(model_path) then
|
||||||
local scale_model_path = path.join(opt.model_dir, ("scale%.1fx_model.t7"):format(opt.scale))
|
local scale_model_path = path.join(opt.model_dir, ("scale%.1fx_model.t7"):format(opt.scale))
|
||||||
local t, scale_model = pcall(load_model, scale_model_path, opt.force_cudnn)
|
local t, scale_model = pcall(w2nn.load_model, scale_model_path, opt.force_cudnn)
|
||||||
local model = w2nn.load_model(model_path, opt.force_cudnn)
|
local model = w2nn.load_model(model_path, opt.force_cudnn)
|
||||||
if not t then
|
if not t then
|
||||||
scale_model = model
|
scale_model = model
|
||||||
|
@ -165,7 +165,7 @@ local function convert_frames(opt)
|
||||||
if path.exists(model_path) then
|
if path.exists(model_path) then
|
||||||
noise_scale_model[opt.noise_level] = w2nn.load_model(model_path, opt.force_cudnn)
|
noise_scale_model[opt.noise_level] = w2nn.load_model(model_path, opt.force_cudnn)
|
||||||
model_path = path.join(opt.model_dir, ("scale%.1fx_model.t7"):format(opt.scale))
|
model_path = path.join(opt.model_dir, ("scale%.1fx_model.t7"):format(opt.scale))
|
||||||
t, scale_model = pcall(load_model, model_path, opt.force_cudnn)
|
t, scale_model = pcall(w2nn.load_model, model_path, opt.force_cudnn)
|
||||||
if not t then
|
if not t then
|
||||||
scale_model = noise_scale_model[opt.noise_level]
|
scale_model = noise_scale_model[opt.noise_level]
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in a new issue