Fix clampImage function call to saturate

1 files changed, 2 insertions, 2 deletions

diff --git a/init.lua b/init.lua
index f1658cf..8c0ca0d 100644
--- a/init.lua
+++ b/init.lua
@@ -143,8 +143,8 @@ local function clampImage(tensor)
       return tensor
    end
    local a = torch.Tensor():resize(tensor:size()):copy(tensor)
-   a.image.saturate(tensor) -- bound btwn 0 and 1
-   a:mul(255)               -- remap to [0..255]
+   a.image.saturate(a) -- bound btwn 0 and 1
+   a:mul(255)          -- remap to [0..255]
    return a
 end