From 08c4967afb779ba95cd68dfc7fc763276a4d29c8 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Sun, 23 May 2021 01:29:06 -0700 Subject: [PATCH] laplace noise --- samplelib/SampleProcessor.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index c20f628..421526d 100644 --- a/samplelib/SampleProcessor.py +++ b/samplelib/SampleProcessor.py @@ -227,12 +227,13 @@ class SampleProcessor(object): noise_scale = (20 * np.random.random() + 20) / 255.0 if noise_type == 'gaussian': - noise = np.random.normal(scale=noise_scale, size=img.shape) - img += noise - elif noise_type == 'laplace': - # noise = np.random.laplace(scale=noise_scale, size=img.shape) + # noise = np.random.normal(scale=noise_scale, size=img.shape) # img += noise pass + elif noise_type == 'laplace': + noise = np.random.laplace(scale=noise_scale, size=img.shape) + img += noise + # pass elif noise_type == 'poisson': # noise_lam = (15 * np.random.random() + 15) # noise = np.random.poisson(lam=noise_lam, size=img.shape)