|---gaussian
　　　　|---before_images(Dossier d'image avant d'ajouter du bruit)
　　　　|---before_images(Dossier d'image après l'ajout de bruit)
　　　　|---gaussian.py

# -*- coding: utf-8 -*-

import cv2
import numpy as np
import sys
import os
import glob

#Bruit gaussien
def addGaussianNoise(src):
    row,col,ch= src.shape
    mean = 0
    var = 0.1
    sigma = 15
    gauss = np.random.normal(mean,sigma,(row,col,ch))
    gauss = gauss.reshape(row,col,ch)
    noisy = src + gauss
    
    return noisy

#Substitution du répertoire où réside le programme
current_dir = os.getcwd()
#Substitution du répertoire dans lequel l'image existe
before_images = glob.glob(current_dir + "\\before_images\\*") 

i = 0

for image in before_images:
    if image == current_dir + "\\before_images\\Thumbs.db":
        continue
    else:
        #Chargement des images
        img = cv2.imread(image)
        
        #Ajoute du bruit
        after_image = addGaussianNoise(img)
        
        #Enregistrer l'image
        cv2.imwrite(current_dir + '\\after_images\\' + str(i) + '.jpg', after_image) 
        i += 1

Recommended Posts