Skip to content

Instantly share code, notes, and snippets.

@ndinh215

ndinh215/add_gaussian_noise.py

Forked from Prasad9/add_gaussian_noise.py
Created August 14, 2022 09:52
Show Gist options
  • Save ndinh215/53d4bf7e08031dc43b3e549f1dccf5dc to your computer and use it in GitHub Desktop.
Save ndinh215/53d4bf7e08031dc43b3e549f1dccf5dc to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. @Prasad9 Prasad9 revised this gist Oct 21, 2017. 1 changed file with 1 addition and 1 deletion.
    2 changes: 1 addition & 1 deletion add_gaussian_noise.py
    View file
    Original file line number Diff line number Diff line change
    @@ -16,4 +16,4 @@ def add_gaussian_noise(X_imgs):
    gaussian_noise_imgs = np.array(gaussian_noise_imgs, dtype = np.float32)
    return gaussian_noise_imgs

    gaussian_noise_imgs = add_gaussian_noise(X_data)
    gaussian_noise_imgs = add_gaussian_noise(X_imgs)
  2. @Prasad9 Prasad9 created this gist Oct 21, 2017.
    19 changes: 19 additions & 0 deletions add_gaussian_noise.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,19 @@
    import cv2

    def add_gaussian_noise(X_imgs):
    gaussian_noise_imgs = []
    row, col, _ = X_imgs[0].shape
    # Gaussian distribution parameters
    mean = 0
    var = 0.1
    sigma = var ** 0.5

    for X_img in X_imgs:
    gaussian = np.random.random((row, col, 1)).astype(np.float32)
    gaussian = np.concatenate((gaussian, gaussian, gaussian), axis = 2)
    gaussian_img = cv2.addWeighted(X_img, 0.75, 0.25 * gaussian, 0.25, 0)
    gaussian_noise_imgs.append(gaussian_img)
    gaussian_noise_imgs = np.array(gaussian_noise_imgs, dtype = np.float32)
    return gaussian_noise_imgs

    gaussian_noise_imgs = add_gaussian_noise(X_data)