|
@@ -4,7 +4,7 @@ require 'pl'
|
|
|
require './lib/LeakyReLU'
|
|
|
|
|
|
local iproc = require './lib/iproc'
|
|
|
-local reconstract = require './lib/reconstract'
|
|
|
+local reconstruct = require './lib/reconstruct'
|
|
|
local image_loader = require './lib/image_loader'
|
|
|
|
|
|
local BLOCK_OFFSET = 7
|
|
@@ -37,12 +37,12 @@ local function waifu2x()
|
|
|
local model = torch.load(path.join(opt.model_dir,
|
|
|
("noise%d_model.t7"):format(opt.noise_level)), "ascii")
|
|
|
model:evaluate()
|
|
|
- new_x = reconstract(model, x, BLOCK_OFFSET)
|
|
|
+ new_x = reconstruct(model, x, BLOCK_OFFSET)
|
|
|
elseif opt.m == "scale" then
|
|
|
local model = torch.load(path.join(opt.model_dir, "scale2.0x_model.t7"), "ascii")
|
|
|
model:evaluate()
|
|
|
x = iproc.scale(x, x:size(3) * 2.0, x:size(2) * 2.0)
|
|
|
- new_x = reconstract(model, x, BLOCK_OFFSET)
|
|
|
+ new_x = reconstruct(model, x, BLOCK_OFFSET)
|
|
|
elseif opt.m == "noise_scale" then
|
|
|
local noise_model = torch.load(path.join(opt.model_dir,
|
|
|
("noise%d_model.t7"):format(opt.noise_level)), "ascii")
|
|
@@ -50,9 +50,9 @@ local function waifu2x()
|
|
|
|
|
|
noise_model:evaluate()
|
|
|
scale_model:evaluate()
|
|
|
- x = reconstract(noise_model, x, BLOCK_OFFSET)
|
|
|
+ x = reconstruct(noise_model, x, BLOCK_OFFSET)
|
|
|
x = iproc.scale(x, x:size(3) * 2.0, x:size(2) * 2.0)
|
|
|
- new_x = reconstract(scale_model, x, BLOCK_OFFSET)
|
|
|
+ new_x = reconstruct(scale_model, x, BLOCK_OFFSET)
|
|
|
else
|
|
|
error("undefined method:" .. opt.method)
|
|
|
end
|