OrgLion-ML
/
openface
mirror of https://github.com/cmusatyalab/openface.git
1
0
Fork 0
Code Issues Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
batch-represent/dataset.lua
View File

@ -328,7 +328,7 @@ end
-- by default, just load the image and return it -- by default, just load the image and return it
function dataset:defaultSampleHook(imgpath) 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]) out = image.scale(out, self.sampleSize[3], self.sampleSize[2])
return out return out
end end

2
training/dataset.lua
View File

@ -330,7 +330,7 @@ end
-- by default, just load the image and return it -- by default, just load the image and return it
function dataset:defaultSampleHook(imgpath) 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]) out = image.scale(out, self.sampleSize[3], self.sampleSize[2])
return out return out
end end