diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index f36c559..f7d7f17 100644 --- a/samplelib/SampleProcessor.py +++ b/samplelib/SampleProcessor.py @@ -227,13 +227,13 @@ class SampleProcessor(object): noise_scale = (20 * np.random.random() + 20) if noise_type == 'gaussian': - # noise = np.random.normal(scale=noise_scale, size=img.shape) - # img += noise - pass + 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 + 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)