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

Change downscaling filters when photo training

Browse source
This commit is contained in:
nagadomi 2015-11-30 17:19:50 +09:00
parent aaac6ed6e5
commit a975155828
1 changed files with 17 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
lib/pairwise_transform.lua
View file

@ -74,16 +74,23 @@ local function active_cropping(x, y, size, p, tries)
end
end
function pairwise_transform.scale(src, scale, size, offset, n, options)
local filters = {
"Box","Box", -- 0.012756949974688
"Blackman", -- 0.013191924552285
--"Cartom", -- 0.013753536746706
--"Hanning", -- 0.013761314529647
--"Hermite", -- 0.013850225205266
"Sinc", -- 0.014095824314306
"Lanczos", -- 0.014244299255442
"Catrom"
}
local filters;
if options.style == "photo" then
filters = {
"Box", "lanczos", "Catrom"
}
else
filters = {
"Box","Box", -- 0.012756949974688
"Blackman", -- 0.013191924552285
--"Catrom", -- 0.013753536746706
--"Hanning", -- 0.013761314529647
--"Hermite", -- 0.013850225205266
"Sinc", -- 0.014095824314306
"Lanczos", -- 0.014244299255442
}
end
local unstable_region_offset = 8
local downscale_filter = filters[torch.random(1, #filters)]
local y = preprocess(src, size, options)