|---gaussian
　　　　|---before_images(Bildordner vor dem Hinzufügen von Rauschen)
　　　　|---before_images(Bildordner nach dem Hinzufügen von Rauschen)
　　　　|---gaussian.py

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

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

#Gaußsches Rauschen
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

#Ersetzen des Verzeichnisses, in dem sich das Programm befindet
current_dir = os.getcwd()
#Ersetzen des Verzeichnisses, in dem das Bild vorhanden ist
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:
        #Bilder laden
        img = cv2.imread(image)
        
        #Rauschen hinzufügen
        after_image = addGaussianNoise(img)
        
        #Bild speichern
        cv2.imwrite(current_dir + '\\after_images\\' + str(i) + '.jpg', after_image) 
        i += 1

Recommended Posts