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

Fix the data type image.load 

Thanks Daniil Belkov!

`byte` was defaulting to `"float"`
This commit is contained in:
Brandon Amos 2016-10-07 14:27:13 -04:00
parent 041c90f0c1
commit 1267e1abb3
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

@ -327,7 +327,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, 3, byte) local out = image.load(imgpath, 3, 'float')
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-issue-132.lua
View File

@ -272,7 +272,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, 3, byte) local out = image.load(imgpath, 3, 'float')
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