From 6856b5ad22736529c64e1787be03247121b755af Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Sun, 23 May 2021 01:40:42 -0700 Subject: [PATCH] poisson noise --- samplelib/SampleProcessor.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index 421526d..f36c559 100644 --- a/samplelib/SampleProcessor.py +++ b/samplelib/SampleProcessor.py @@ -224,21 +224,21 @@ class SampleProcessor(object): # Apply random noise if random_noise: noise_type = np.random.choice(['gaussian', 'laplace', 'poisson']) - noise_scale = (20 * np.random.random() + 20) / 255.0 + noise_scale = (20 * np.random.random() + 20) if noise_type == 'gaussian': # 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) + # 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) + img += noise / 255.0 + # pass img = imagelib.warp_by_params (params_per_resolution[resolution], img, warp, transform, can_flip=True, border_replicate=border_replicate) img = np.clip(img.astype(np.float32), 0, 1)