From 5cda0b71d7219f5bbe060ceef2a708ae05b78ff4 Mon Sep 17 00:00:00 2001 From: nagadomi Date: Tue, 1 Dec 2015 20:42:44 +0900 Subject: [PATCH] Fix weird value in alpha channel --- lib/image_loader.lua | 2 ++ 1 file changed, 2 insertions(+) diff --git a/lib/image_loader.lua b/lib/image_loader.lua index b5717b8..48f92a2 100644 --- a/lib/image_loader.lua +++ b/lib/image_loader.lua @@ -16,10 +16,12 @@ function image_loader.encode_png(rgb, depth) rgb = rgb:clone():add(clip_eps8) rgb[torch.lt(rgb, 0.0)] = 0.0 rgb[torch.gt(rgb, 1.0)] = 1.0 + rgb = rgb:mul(255):long():float():div(255) else rgb = rgb:clone():add(clip_eps16) rgb[torch.lt(rgb, 0.0)] = 0.0 rgb[torch.gt(rgb, 1.0)] = 1.0 + rgb = rgb:mul(65535):long():float():div(65535) end local im if rgb:size(1) == 4 then -- RGBA