소스 검색

fix image size validation in convert.lua

nagadomi 10 년 전
부모
커밋
99db05a753
1개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  1. 2 1
      convert_data.lua

+ 2 - 1
convert_data.lua

@@ -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