diff --git a/lib/pairwise_transform_jpeg.lua b/lib/pairwise_transform_jpeg.lua index 86848ff..4d05dca 100644 --- a/lib/pairwise_transform_jpeg.lua +++ b/lib/pairwise_transform_jpeg.lua @@ -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 diff --git a/lib/pairwise_transform_jpeg_scale.lua b/lib/pairwise_transform_jpeg_scale.lua index 2f4b86c..d46ded0 100644 --- a/lib/pairwise_transform_jpeg_scale.lua +++ b/lib/pairwise_transform_jpeg_scale.lua @@ -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