images_agumentation_rotation.py

from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
'''img = Image.open('//home//harish//Desktop//videosforabg//dog1.jpg')
img = np.array(img)
plt.imshow(img)
plt.show()

flipped_img = np.fliplr(img)
plt.imshow(flipped_img)
plt.show()
'''


import numpy as np
import random
import cv2

def sp_noise(image,prob):
    '''
    Add salt and pepper noise to image
    prob: Probability of the noise
    '''
    output = np.zeros(image.shape,np.uint8)
    thres = 1 - prob 
    for i in range(image.shape[0]):
        for j in range(image.shape[1]):
            rdn = random.random()
            if rdn < prob:
                output[i][j] = 0
            elif rdn > thres:
                output[i][j] = 255
            else:
                output[i][j] = image[i][j]
    return output

image = cv2.imread('/home/harish/Desktop/videosforabg/output1125.jpg',0) # Only for grayscale image
noise_img = sp_noise(image,0.05)
cv2.imwrite('sp_noise.jpg', noise_img)
plt.show()