CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-12 16:04:31 +12:00
Code Issues Releases Wiki Activity

fix image size validation in convert.lua

Browse source
This commit is contained in:
nagadomi 2015-07-11 21:53:45 +09:00
parent d8ca857a82
commit 99db05a753
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
convert_data.lua
View file

@ -21,6 +21,7 @@ local function crop_4x(x)
end end
local function load_images(list) local function load_images(list)
local MARGIN = 32
local count = count_lines(list) local count = count_lines(list)
local fp = io.open(list, "r") local fp = io.open(list, "r")
local x = {} local x = {}
@ -28,7 +29,7 @@ local function load_images(list)
for line in fp:lines() do for line in fp:lines() do
local im = crop_4x(image_loader.load_byte(line)) local im = crop_4x(image_loader.load_byte(line))
if im then if im then
if im:size(2) >= settings.crop_size * 2 and im:size(3) >= settings.crop_size * 2 then if im:size(2) > (settings.crop_size * 2 + MARGIN) and im:size(3) > (settings.crop_size * 2 + MARGIN) then
table.insert(x, im) table.insert(x, im)
end end
else else