From 4a3d1d971e9a3ab3317bf71ff8f3dbd50787fd32 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Sun, 23 May 2021 01:51:19 -0700 Subject: [PATCH] Fix scaling --- samplelib/SampleProcessor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index f7d7f17..2e47fc9 100644 --- a/samplelib/SampleProcessor.py +++ b/samplelib/SampleProcessor.py @@ -228,11 +228,11 @@ class SampleProcessor(object): if noise_type == 'gaussian': noise = np.random.normal(scale=noise_scale, size=img.shape) - img += noise + img += noise / 255.0 # pass elif noise_type == 'laplace': noise = np.random.laplace(scale=noise_scale, size=img.shape) - img += noise + img += noise / 255.0 # pass elif noise_type == 'poisson': noise_lam = (15 * np.random.random() + 15)