CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-25 15:19:33 +12:00
Code Issues Releases Wiki Activity

Add diff method to tools/benchmark

Browse source
This commit is contained in:
nagadomi 2016-08-26 06:34:23 +09:00
parent 8d77d6e2e9
commit 83a8cbe3bd
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tools/benchmark.lua
View file

@ -18,7 +18,7 @@ cmd:option("-dir", "./data/test", 'test image directory')
cmd:option("-file", "", 'test image file list')
cmd:option("-model1_dir", "./models/anime_style_art_rgb", 'model1 directory')
cmd:option("-model2_dir", "", 'model2 directory (optional)')
cmd:option("-method", "scale", '(scale|noise|noise_scale|user)')
cmd:option("-method", "scale", '(scale|noise|noise_scale|user|diff)')
cmd:option("-filter", "Catrom", "downscaling filter (Box|Lanczos|Catrom(Bicubic))")
cmd:option("-resize_blur", 1.0, 'blur parameter for resize')
cmd:option("-color", "y", '(rgb|y|r|g|b)')
@ -343,6 +343,10 @@ local function benchmark(opt, x, model1, model2)
model2_time = model2_time + (sys.clock() - t)
end
end
elseif opt.method == "diff" then
input = x[i].x
ground_truth = x[i].y
model1_output = input
end
mse = MSE(ground_truth, model1_output, opt.color)
model1_mse = model1_mse + mse
@ -618,4 +622,7 @@ elseif opt.method == "user" then
end
local test = load_user_data(opt.y_dir, opt.y_file, opt.x_dir, opt.x_file)
benchmark(opt, test, model1, model2)
elseif opt.method == "diff" then
local test = load_user_data(opt.y_dir, opt.y_file, opt.x_dir, opt.x_file)
benchmark(opt, test, nil, nil)
end