CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-01 10:39:30 +12:00
Code Issues Releases Wiki Activity

Fix unused crop_size option

Browse source
This commit is contained in:
nagadomi 2015-11-09 04:10:51 +09:00
parent b335f3a9ad
commit c21516f04d
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
waifu2x.lua
View file

@ -53,7 +53,7 @@ local function convert_image(opt)
if not scale_model then
error("Load Error: " .. scale_model_path)
end
x = image_f(noise_model, x)
x = image_f(noise_model, x, opt.crop_size)
new_x = scale_f(scale_model, opt.scale, x, opt.crop_size)
else
error("undefined method:" .. opt.method)
@ -126,11 +126,11 @@ local function convert_frames(opt)
if opt.m == "noise" and opt.noise_level == 1 then
new_x = image_f(noise1_model, x, opt.crop_size)
elseif opt.m == "noise" and opt.noise_level == 2 then
new_x = image_func(noise2_model, x)
new_x = image_func(noise2_model, x, opt.crop_size)
elseif opt.m == "scale" then
new_x = scale_f(scale_model, opt.scale, x, opt.crop_size)
elseif opt.m == "noise_scale" and opt.noise_level == 1 then
x = image_f(noise1_model, x)
x = image_f(noise1_model, x, opt.crop_size)
new_x = scale_f(scale_model, opt.scale, x, opt.crop_size)
elseif opt.m == "noise_scale" and opt.noise_level == 2 then
x = image_f(noise2_model, x, opt.crop_size)