from pylab import *
from matplotlib import *
import numpy

def rgb2gray(rgb):
	r,g,b=rgb[:, :, 0], rgb[:,:,1], rgb[:,:,2]
	gray=0.2989*r+0.5870*g+0.1140*b
	return gray

image = imread("oko.jpg")
gimage=rgb2gray(image)
imshow(gimage, cmap="gray")
title("Obraz podstawowy")
show()

szum=rand(gimage.shape[0],gimage.shape[1])
szum=szum*100
imshow(szum, cmap="gray")
title("Szum o rozkladzie rownomiernym")
show()

gimageszum = gimage + szum
imshow(gimageszum, cmap="gray")
title("Obraz z szumem o rozkladzie rownomiernym")
show()

szumGauss=randn(gimage.shape[0],gimage.shape[1])
szumGauss=szumGauss * 100
imshow(szumGauss, cmap="gray")
title("Szum Gaussa")
show()

gimageszumGauss= gimage + szumGauss
imshow(gimageszumGauss, cmap="gray")
title("Obraz z szumem Gaussa")
show()

sol_pieprz = gimage
for i in range(8000):
sol_pieprz[randint(0,sol_pieprz.shape[0]),randint(0,sol_pieprz.shape[1])] =1
sol_pieprz[randint(0,sol_pieprz.shape[0]),randint(0,sol_pieprz.shape[1])] =0
imshow(sol_pieprz, cmap="gray")
title("Obraz z szumem pieprz i sol")
show()