CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-26 18:20:26 +12:00
Code Issues Releases Wiki Activity

load float images directly

Browse source
This commit is contained in:
nagadomi 2015-11-21 00:58:30 +09:00
parent c47df93505
commit e9187ecf1c
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tools/benchmark.lua
View file

@ -59,6 +59,12 @@ local function transform_jpeg(x, opt)
end
return x
end
local function baseline_scale(x, filter)
return iproc.scale(x,
x:size(3) * 2.0,
x:size(2) * 2.0,
filter)
end
local function transform_scale(x, opt)
return iproc.scale(x,
x:size(3) * 0.5,
@ -79,9 +85,6 @@ local function benchmark(opt, x, input_func, model1, model2)
local input, model1_output, model2_output, baseline_output
input = input_func(ground_truth, opt)
input = input:float():div(255)
ground_truth = ground_truth:float():div(255)
t = sys.clock()
if input:size(3) == ground_truth:size(3) then
model1_output = reconstruct.image(model1, input)
@ -93,7 +96,7 @@ local function benchmark(opt, x, input_func, model1, model2)
if model2 then
model2_output = reconstruct.scale(model2, 2.0, input)
end
baseline_output = iproc.scale(input, input:size(3) * 2, input:size(2) * 2, opt.filter)
baseline_output = baseline_scale(input, opt.filter)
end
if opt.color == "y" then
model1_mse = model1_mse + YMSE(ground_truth, model1_output)
@ -162,7 +165,7 @@ local function load_data(test_dir)
local test_x = {}
local files = dir.getfiles(test_dir, "*.*")
for i = 1, #files do
table.insert(test_x, iproc.crop_mod4(image_loader.load_byte(files[i])))
table.insert(test_x, iproc.crop_mod4(image_loader.load_float(files[i])))
xlua.progress(i, #files)
end
return test_x