Correct calls to image.load

This commit is contained in:
Brandon Amos 2015-10-20 11:34:52 -04:00
parent 0771ba87ba
commit 1fb67767b7
2 changed files with 2 additions and 2 deletions

View File

@ -328,7 +328,7 @@ end
-- by default, just load the image and return it
function dataset:defaultSampleHook(imgpath)
local out = image.load(imgpath, self.loadSize[1])
local out = image.load(imgpath, 3, byte)
out = image.scale(out, self.sampleSize[3], self.sampleSize[2])
return out
end

View File

@ -330,7 +330,7 @@ end
-- by default, just load the image and return it
function dataset:defaultSampleHook(imgpath)
local out = image.load(imgpath, self.loadSize[1])
local out = image.load(imgpath, 3, byte)
out = image.scale(out, self.sampleSize[3], self.sampleSize[2])
return out
end