Image loading: Use 'float' instead of an undefined variable.
This commit is contained in:
parent
60d31c09b4
commit
13701ec424
|
@ -61,7 +61,7 @@ while true do
|
|||
-- as a CSV.
|
||||
local imgPath = io.read("*line")
|
||||
if imgPath and imgPath:len() ~= 0 then
|
||||
img[1] = image.load(imgPath, 3, byte)
|
||||
img[1] = image.load(imgPath, 3, 'float')
|
||||
img[1] = image.scale(img[1], opt.imgDim, opt.imgDim)
|
||||
local rep
|
||||
if opt.cuda then
|
||||
|
|
Loading…
Reference in New Issue