CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-28 19:20:32 +12:00
Code Issues Releases Wiki Activity

Fix a block_size issue when using upconv_7 model

Browse source
This commit is contained in:
nagadomi 2016-06-12 05:15:24 +09:00
parent 9103d393fe
commit 6be1479710
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lib/reconstruct.lua
View file

@ -9,8 +9,8 @@ local function reconstruct_nn(model, x, inner_scale, offset, block_size, batch_s
end
local ch = x:size(1)
local new_x = torch.Tensor(x:size(1), x:size(2) * inner_scale, x:size(3) * inner_scale):zero()
local input_block_size = block_size / inner_scale
local output_block_size = block_size
local input_block_size = block_size
local output_block_size = block_size * inner_scale
local output_size = output_block_size - offset * 2
local output_size_in_input = input_block_size - math.ceil(offset / inner_scale) * 2
local input_indexes = {}
@ -81,7 +81,7 @@ local function padding_params(x, model, block_size)
p.x_h = x:size(2)
p.inner_scale = reconstruct.inner_scale(model)
local input_offset = math.ceil(offset / p.inner_scale)
local input_block_size = block_size / p.inner_scale
local input_block_size = block_size
local process_size = input_block_size - input_offset * 2
local h_blocks = math.floor(p.x_h / process_size) +
((p.x_h % process_size == 0 and 0) or 1)