Fix the data type image.load

Thanks Daniil Belkov! `byte` was defaulting to `"float"`
cmusatyalab · Oct 7, 2016 · 1267e1a · 1267e1a
1 parent 041c90f
commit 1267e1a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/batch-represent/dataset.lua b/batch-represent/dataset.lua
@@ -327,7 +327,7 @@ end
 
 -- by default, just load the image and return it
 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])
    return out
 end

diff --git a/training/dataset-issue-132.lua b/training/dataset-issue-132.lua
@@ -272,7 +272,7 @@ end
 
 -- by default, just load the image and return it
 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])
    return out
 end