فهرست منبع

Fix weird value in alpha channel

nagadomi 9 سال پیش
والد
کامیت
5cda0b71d7
1فایلهای تغییر یافته به همراه2 افزوده شده و 0 حذف شده
  1. 2 0
      lib/image_loader.lua

+ 2 - 0
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