From 99db05a75339907d43f18287d1380b5efa1e0935 Mon Sep 17 00:00:00 2001 From: nagadomi Date: Sat, 11 Jul 2015 21:53:45 +0900 Subject: [PATCH] fix image size validation in convert.lua --- convert_data.lua | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/convert_data.lua b/convert_data.lua index 0fb9833..dbd9978 100644 --- a/convert_data.lua +++ b/convert_data.lua @@ -21,6 +21,7 @@ local function crop_4x(x) end local function load_images(list) + local MARGIN = 32 local count = count_lines(list) local fp = io.open(list, "r") local x = {} @@ -28,7 +29,7 @@ local function load_images(list) for line in fp:lines() do local im = crop_4x(image_loader.load_byte(line)) 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) end else