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

Add checking for malformed images

Browse source
This commit is contained in:
nagadomi 2016-12-05 10:37:42 +09:00
parent d2cfb8f104
commit a47e0a9804
1 changed files with 35 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
convert_data.lua
View file

@ -82,6 +82,7 @@ local function load_images(list)
local skip = false
local alpha_color = torch.random(0, 1)
if im then
if meta and meta.alpha then
if settings.use_transparent_png then
im = alpha_util.fill(im, meta.alpha, alpha_color)
@ -99,12 +100,16 @@ local function load_images(list)
-- method == user
local yy = im
local xx, meta2 = image_loader.load_byte(csv_meta.x)
if xx then
if meta2 and meta2.alpha then
xx = alpha_util.fill(xx, meta2.alpha, alpha_color)
end
xx, yy = crop_if_large_pair(xx, yy, settings.max_training_image_size)
table.insert(x, {{y = compression.compress(yy), x = compression.compress(xx)},
{data = {filters = filters, has_x = true}}})
else
io.stderr:write(string.format("\n%s: skip: load error.\n", csv_meta.x))
end
else
im = crop_if_large(im, settings.max_training_image_size)
im = iproc.crop_mod4(im)
@ -112,17 +117,16 @@ local function load_images(list)
if settings.random_half_rate > 0.0 then
scale = 2.0
end
if im then
if im:size(2) > (settings.crop_size * scale + MARGIN) and im:size(3) > (settings.crop_size * scale + MARGIN) then
table.insert(x, {compression.compress(im), {data = {filters = filters}}})
else
io.stderr:write(string.format("\n%s: skip: image is too small (%d > size).\n", filename, settings.crop_size * scale + MARGIN))
end
end
end
else
io.stderr:write(string.format("\n%s: skip: load error.\n", filename))
end
end
end
xlua.progress(i, #csv)
if i % 10 == 0 then
collectgarbage()