瀏覽代碼

Fix a bug in noise reduction when using Y model

nagadomi 9 年之前
父節點
當前提交
deae802267
共有 1 個文件被更改,包括 3 次插入1 次删除
  1. 3 1
      lib/reconstruct.lua

+ 3 - 1
lib/reconstruct.lua

@@ -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)