From 28cc864c0ceadc303339ec323b5c61a6efc71c91 Mon Sep 17 00:00:00 2001
From: nagadomi <nagadomi@nurs.or.jp>
Date: Thu, 12 Nov 2015 08:03:24 +0900
Subject: [PATCH] Fix NaN bug in iproc.white_noise

---
 lib/iproc.lua | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/lib/iproc.lua b/lib/iproc.lua
index d37ddcc..2ea20c4 100644
--- a/lib/iproc.lua
+++ b/lib/iproc.lua
@@ -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