noise0
This commit is contained in:
parent
cbe1874be3
commit
8e90a44d30
|
@ -60,7 +60,10 @@ function pairwise_transform.jpeg_(src, quality, size, offset, n, options)
|
||||||
end
|
end
|
||||||
function pairwise_transform.jpeg(src, style, level, size, offset, n, options)
|
function pairwise_transform.jpeg(src, style, level, size, offset, n, options)
|
||||||
if style == "art" then
|
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)},
|
return pairwise_transform.jpeg_(src, {torch.random(65, 85)},
|
||||||
size, offset, n, options)
|
size, offset, n, options)
|
||||||
elseif level == 2 or level == 3 then
|
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)
|
error("unknown noise level: " .. level)
|
||||||
end
|
end
|
||||||
elseif style == "photo" then
|
elseif style == "photo" then
|
||||||
-- level adjusting by -nr_rate
|
if level == 0 then
|
||||||
return pairwise_transform.jpeg_(src, {torch.random(30, 70)},
|
return pairwise_transform.jpeg_(src, {torch.random(85, 95)},
|
||||||
size, offset, n,
|
size, offset, n,
|
||||||
options)
|
options)
|
||||||
|
else
|
||||||
|
return pairwise_transform.jpeg_(src, {torch.random(37, 70)},
|
||||||
|
size, offset, n,
|
||||||
|
options)
|
||||||
|
end
|
||||||
else
|
else
|
||||||
error("unknown style: " .. style)
|
error("unknown style: " .. style)
|
||||||
end
|
end
|
||||||
|
|
|
@ -23,7 +23,9 @@ end
|
||||||
|
|
||||||
local function add_jpeg_noise(src, style, level, options)
|
local function add_jpeg_noise(src, style, level, options)
|
||||||
if style == "art" then
|
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)
|
return add_jpeg_noise_(src, {torch.random(65, 85)}, options)
|
||||||
elseif level == 2 or level == 3 then
|
elseif level == 2 or level == 3 then
|
||||||
-- level 2/3 adjusting by -nr_rate. for level3, -nr_rate=1
|
-- 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)
|
error("unknown noise level: " .. level)
|
||||||
end
|
end
|
||||||
elseif style == "photo" then
|
elseif style == "photo" then
|
||||||
-- level adjusting by -nr_rate
|
if level == 0 then
|
||||||
return add_jpeg_noise_(src, {torch.random(30, 70)}, options)
|
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
|
else
|
||||||
error("unknown style: " .. style)
|
error("unknown style: " .. style)
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in a new issue