mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-08-19 04:59:27 -07:00
all noise enabled
This commit is contained in:
parent
6856b5ad22
commit
411346b2bd
1 changed files with 6 additions and 6 deletions
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue