CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-19 20:32:22 +12:00
Code Issues Releases Wiki Activity

Fix a bug that the alpha channel is upscaled by noise_scale model

Browse source
This commit is contained in:
nagadomi 2016-07-14 23:38:04 +09:00
parent 651e8f8ecb
commit be67a4e001
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
waifu2x.lua
View file

@ -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))
if path.exists(model_path) then
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)
if not t then
scale_model = model
@ -165,7 +165,7 @@ local function convert_frames(opt)
if path.exists(model_path) then
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))
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
scale_model = noise_scale_model[opt.noise_level]
end