From 1fb67767b75800bb369ac17309f8d77702b88c6c Mon Sep 17 00:00:00 2001 From: Brandon Amos Date: Tue, 20 Oct 2015 11:34:52 -0400 Subject: [PATCH] Correct calls to image.load --- batch-represent/dataset.lua | 2 +- training/dataset.lua | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/batch-represent/dataset.lua b/batch-represent/dataset.lua index f365317..fc1d13a 100644 --- a/batch-represent/dataset.lua +++ b/batch-represent/dataset.lua @@ -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 diff --git a/training/dataset.lua b/training/dataset.lua index 34df663..e5a6eb0 100644 --- a/training/dataset.lua +++ b/training/dataset.lua @@ -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