1
0
Fork 0
mirror of synced 2024-06-13 08:24:30 +12:00

Fix NaN bug in iproc.white_noise

This commit is contained in:
nagadomi 2015-11-12 08:03:24 +09:00
parent dcfb0d170d
commit 28cc864c0c

View file

@ -99,7 +99,10 @@ function iproc.white_noise(src, std, rgb_weights, gamma)
local dest
if gamma ~= 0 then
dest = src:clone():pow(gamma):add(noise):pow(1.0 / gamma)
dest = src:clone():pow(gamma):add(noise)
dest[torch.lt(dest, 0.0)] = 0.0
dest[torch.gt(dest, 1.0)] = 1.0
dest:pow(1.0 / gamma)
else
dest = src + noise
end