CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-01 10:39:30 +12:00
Code Issues Releases Wiki Activity

refactor

Browse source
This commit is contained in:
nagadomi 2015-11-09 04:35:11 +09:00
parent c21516f04d
commit af1b9c604b
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
lib/image_loader.lua
View file

@ -4,8 +4,8 @@ require 'pl'
local image_loader = {}
local clip_eta8 = (1.0 / 255.0) * 0.5 - (1.0e-7 * (1.0 / 255.0) * 0.5)
local clip_eta16 = (1.0 / 65535.0) * 0.5 - (1.0e-7 * (1.0 / 65535.0) * 0.5)
local clip_eps8 = (1.0 / 255.0) * 0.5 - (1.0e-7 * (1.0 / 255.0) * 0.5)
local clip_eps16 = (1.0 / 65535.0) * 0.5 - (1.0e-7 * (1.0 / 65535.0) * 0.5)
local background_color = 0.5
function image_loader.decode_float(blob)
@ -31,11 +31,11 @@ function image_loader.encode_png(rgb, alpha, depth)
rgba[4]:copy(alpha)
if depth < 16 then
rgba:add(clip_eta8)
rgba:add(clip_eps8)
rgba[torch.lt(rgba, 0.0)] = 0.0
rgba[torch.gt(rgba, 1.0)] = 1.0
else
rgba:add(clip_eta16)
rgba:add(clip_eps16)
rgba[torch.lt(rgba, 0.0)] = 0.0
rgba[torch.gt(rgba, 1.0)] = 1.0
end
@ -43,11 +43,11 @@ function image_loader.encode_png(rgb, alpha, depth)
return im:depth(depth):format("PNG"):toBlob(9)
else
if depth < 16 then
rgb = rgb:clone():add(clip_eta8)
rgb = rgb:clone():add(clip_eps8)
rgb[torch.lt(rgb, 0.0)] = 0.0
rgb[torch.gt(rgb, 1.0)] = 1.0
else
rgb = rgb:clone():add(clip_eta16)
rgb = rgb:clone():add(clip_eps16)
rgb[torch.lt(rgb, 0.0)] = 0.0
rgb[torch.gt(rgb, 1.0)] = 1.0
end