CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-16 19:02:21 +12:00
Code Issues Releases Wiki Activity

Fix zero division error in jaccard

Browse source
This commit is contained in:
nagadomi 2018-04-28 11:04:53 +09:00
parent 0b1a13d9c0
commit e5e73c0ba7
1 changed files with 15 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
tools/benchmark.lua
View file

@ -216,7 +216,12 @@ local function create_metric(metric)
local num_a = ba:sum()
local num_b = bb:sum()
local a_and_b = ba:cmul(bb):sum()
return (a_and_b / (num_a + num_b - a_and_b))
local abab = (num_a + num_b - a_and_b)
if abab > 0 then
return (a_and_b / abab)
else
return 1
end
end}
else
error("unknown metric: " .. metric)
@ -615,10 +620,10 @@ local function load_data_from_dir(test_dir)
table.insert(test_x, {y = iproc.crop_mod4(img),
basename = base})
end
if opt.show_progress then
xlua.progress(i, #files)
end
if i % 10 == 0 then
if opt.show_progress then
xlua.progress(i, #files)
end
collectgarbage()
end
end
@ -636,10 +641,10 @@ local function load_data_from_file(test_file)
table.insert(test_x, {y = iproc.crop_mod4(img),
basename = base})
end
if opt.show_progress then
xlua.progress(i, #files)
end
if i % 10 == 0 then
if opt.show_progress then
xlua.progress(i, #files)
end
collectgarbage()
end
end
@ -694,10 +699,10 @@ local function load_user_data(y_dir, y_file, x_dir, x_file)
x = x,
basename = key})
end
if opt.show_progress then
xlua.progress(i, #y_files)
end
if i % 10 == 0 then
if opt.show_progress then
xlua.progress(i, #y_files)
end
collectgarbage()
end
end