diff --git a/batch-represent/dataset.lua b/batch-represent/dataset.lua index 76a7f4d..7d9cedc 100644 --- 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 index 6b32e38..b396665 100644 --- 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