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

waifu2x.lua

@@ -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