CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-29 03:30:52 +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
end end
function pairwise_transform.scale(src, scale, size, offset, n, options) function pairwise_transform.scale(src, scale, size, offset, n, options)
local filters = { local filters;
"Box","Box", -- 0.012756949974688
"Blackman", -- 0.013191924552285 if options.style == "photo" then
--"Cartom", -- 0.013753536746706 filters = {
--"Hanning", -- 0.013761314529647 "Box", "lanczos", "Catrom"
--"Hermite", -- 0.013850225205266 }
"Sinc", -- 0.014095824314306 else
"Lanczos", -- 0.014244299255442 filters = {
"Catrom" "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 unstable_region_offset = 8
local downscale_filter = filters[torch.random(1, #filters)] local downscale_filter = filters[torch.random(1, #filters)]
local y = preprocess(src, size, options) local y = preprocess(src, size, options)