CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-16 19:02:21 +12:00
Code Issues Releases Wiki Activity

Fix tta-mode

Browse source
This commit is contained in:
nagadomi 2016-10-24 09:06:27 +09:00
parent 61aeb46303
commit f16950438c
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
lib/reconstruct.lua
View file

@ -172,6 +172,9 @@ function reconstruct.scale_rgb(model, scale, x, offset, block_size, batch_size)
return output
end
function reconstruct.image(model, x, block_size)
if model.w2nn_input_size then
block_size = model.w2nn_input_size
end
local i2rgb = false
if x:size(1) == 1 then
local new_x = torch.Tensor(3, x:size(2), x:size(3))
@ -194,6 +197,9 @@ function reconstruct.image(model, x, block_size)
return x
end
function reconstruct.scale(model, scale, x, block_size)
if model.w2nn_input_size then
block_size = model.w2nn_input_size
end
local i2rgb = false
if x:size(1) == 1 then
local new_x = torch.Tensor(3, x:size(2), x:size(3))
@ -287,6 +293,9 @@ local function tta(f, n, model, x, block_size)
return average:div(#augments)
end
function reconstruct.image_tta(model, n, x, block_size)
if model.w2nn_input_size then
block_size = model.w2nn_input_size
end
if reconstruct.is_rgb(model) then
return tta(reconstruct.image_rgb, n, model, x, block_size)
else
@ -294,6 +303,9 @@ function reconstruct.image_tta(model, n, x, block_size)
end
end
function reconstruct.scale_tta(model, n, scale, x, block_size)
if model.w2nn_input_size then
block_size = model.w2nn_input_size
end
if reconstruct.is_rgb(model) then
local f = function (model, x, offset, block_size)
return reconstruct.scale_rgb(model, scale, x, offset, block_size)