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

noise0

Browse source
This commit is contained in:
nagadomi 2016-07-09 09:08:04 +09:00
parent cbe1874be3
commit 8e90a44d30
2 changed files with 22 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
lib/pairwise_transform_jpeg.lua
View file

@ -60,7 +60,10 @@ function pairwise_transform.jpeg_(src, quality, size, offset, n, options)
end
function pairwise_transform.jpeg(src, style, level, size, offset, n, options)
if style == "art" then
if level == 1 then
if level == 0 then
return pairwise_transform.jpeg_(src, {torch.random(85, 95)},
size, offset, n, options)
elseif level == 1 then
return pairwise_transform.jpeg_(src, {torch.random(65, 85)},
size, offset, n, options)
elseif level == 2 or level == 3 then
@ -87,10 +90,15 @@ function pairwise_transform.jpeg(src, style, level, size, offset, n, options)
error("unknown noise level: " .. level)
end
elseif style == "photo" then
-- level adjusting by -nr_rate
return pairwise_transform.jpeg_(src, {torch.random(30, 70)},
size, offset, n,
options)
if level == 0 then
return pairwise_transform.jpeg_(src, {torch.random(85, 95)},
size, offset, n,
options)
else
return pairwise_transform.jpeg_(src, {torch.random(37, 70)},
size, offset, n,
options)
end
else
error("unknown style: " .. style)
end

12
lib/pairwise_transform_jpeg_scale.lua
View file

@ -23,7 +23,9 @@ end
local function add_jpeg_noise(src, style, level, options)
if style == "art" then
if level == 1 then
if level == 0 then
return add_jpeg_noise_(src, {torch.random(85, 95)}, options)
elseif level == 1 then
return add_jpeg_noise_(src, {torch.random(65, 85)}, options)
elseif level == 2 or level == 3 then
-- level 2/3 adjusting by -nr_rate. for level3, -nr_rate=1
@ -44,8 +46,12 @@ local function add_jpeg_noise(src, style, level, options)
error("unknown noise level: " .. level)
end
elseif style == "photo" then
-- level adjusting by -nr_rate
return add_jpeg_noise_(src, {torch.random(30, 70)}, options)
if level == 0 then
return add_jpeg_noise_(src, {torch.random(85, 95)}, options)
else
-- level adjusting by -nr_rate
return add_jpeg_noise_(src, {torch.random(37, 70)}, options)
end
else
error("unknown style: " .. style)
end