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

Fix a bug in noise reduction when using Y model

Browse source
This commit is contained in:
nagadomi 2016-07-06 00:47:58 +09:00
parent edac608f18
commit deae802267
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lib/reconstruct.lua
View file

@ -100,8 +100,10 @@ function reconstruct.image_y(model, x, offset, block_size, batch_size)
local p = padding_params(x, model, block_size)
x = image.rgb2yuv(iproc.padding(x, p.pad_w1, p.pad_w2, p.pad_h1, p.pad_h2))
local y = reconstruct_nn(model, x[1], p.inner_scale, offset, block_size, batch_size)
x = iproc.crop(x, p.pad_w1, p.pad_w2, p.pad_w1 + p.x_w, p.pad_w2 + p.x_h)
x = iproc.crop(x, p.pad_w1, p.pad_h1, p.pad_w1 + p.x_w, p.pad_h1 + p.x_h)
y = iproc.crop(y, 0, 0, p.x_w, p.x_h)
y[torch.lt(y, 0)] = 0
y[torch.gt(y, 1)] = 1
x[1]:copy(y)