Skip to content

Instantly share code, notes, and snippets.

@pepgonzalez

pepgonzalez/Ruido.py

Created February 13, 2013 03:23
Show Gist options
  • Save pepgonzalez/4941998 to your computer and use it in GitHub Desktop.
Save pepgonzalez/4941998 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Ruido.py
from sys import argv
from PIL import Image as I
import random as R
def generarRuido(m, porcentaje):
imagenConRuido = m
p = m.load()
x,y = m.size
totalPixeles = x * y
pixelesMarcados = totalPixeles * porcentaje
pixelesMarcados = pixelesMarcados / 100
print pixelesMarcados
for i in range(int(pixelesMarcados)):
pxx = R.randrange(0,x)
pxy = R.randrange(0,y)
sop = R.uniform(0,1)
pixel = ''
if(sop > .5):
pixel = (255,255,255)
else:
pixel = (0,0,0)
imagenConRuido.putpixel((pxx,pxy),pixel)
imagenConRuido.show()
quitarRuidoPorVecindad(imagenConRuido)
def quitarRuidoPorVecindad(imagen):
m = []
imagen = imagen.convert('RGB')
x,y = imagen.size
pixeles = imagen.load()
c = 1
for i in range(x):
for j in range(y):
px = pixeles[i,j]
b = px[0]
if(b == 255 or b == 0):
m.append((i,j))
c = 0
try:
p1 = pixeles[i+1,j]
c += 1
except:
p1 = (0,0,0)
try:
p2 = pixeles[i-1,j]
c += 1
except:
p2 = (0,0,0)
try:
p3 = pixeles[i,j+1]
c += 1
except:
p3 = (0,0,0)
try:
p4 = pixeles[i,j-1]
c += 1
except:
p4 = (0,0,0)
if(c > 4):
c = 4
r = (p1[0] + p2[0] + p3[0] + p4[0])
g = (p1[0] + p2[0] + p3[0] + p4[0])
b = (p1[0] + p2[0] + p3[0] + p4[0])
pxnvo = (r/c,g/c,b/c)
imagen.putpixel((i,j),pxnvo)
c = 0
#SEGUNdA PASADA
fi = imagen.load()
for e in m:
c = 0
pixel = fi[e]
i,j = e
try:
p1 = fi[i+1,j]
c += 1
except:
p1 = (0,0,0)
try:
p2 = fi[i-1,j]
c += 1
except:
p2 = (0,0,0)
try:
p3 = fi[i,j+1]
c += 1
except:
p3 = (0,0,0)
try:
p4 = fi[i,j-1]
c += 1
except:
p4 = (0,0,0)
r = (p1[0] + p2[0] + p3[0] + p4[0])
g = (p1[0] + p2[0] + p3[0] + p4[0])
b = (p1[0] + p2[0] + p3[0] + p4[0])
pxnvo = (r/c,g/c,b/c)
imagen.putpixel((i,j),pxnvo)
c = 0
imagen.show()
return imagen
def bordesPorResta(im, imagen2):
x,y = im.size
m1 = im.load()
m2 = imagen2.load()
imb = I.new('RGB',(x,y))
for i in range(x):
for j in range(y):
r = m1[i,j][0] - m2[i,j][0]
g = m1[i,j][1] - m2[i,j][1]
b = m1[i,j][2] - m2[i,j][2]
pixel = (r,g,b)
imb.putpixel((i,j),pixel)
imb.show()
def filtroGrisesPromedio(imagen):
x, y = imagen.size
px = imagen.load()
imagenGrises = I.new('RGB',(x,y))
for i in range(x):
for j in range(y):
pixeles = px[i,j]
prom = sum(pixeles) / 3
imagenGrises.putpixel((i,j),(prom,prom,prom))
return imagenGrises
def filtroPromedio(imagen):
x,y = imagen.size
pixeles = imagen.load()
imagenFiltrada = I.new('RGB',(x,y))
c = 1
for i in range(x):
for j in range(y):
px = pixeles[i,j]
try:
p1 = pixeles[i+1,j]
c += 1
except:
p1 = (0,0,0)
try:
p2 = pixeles[i-1,j]
c += 1
except:
p2 = (0,0,0)
try:
p3 = pixeles[i,j+1]
c += 1
except:
p3 = (0,0,0)
try:
p4 = pixeles[i,j-1]
c += 1
except:
p4 = (0,0,0)
if(c > 4):
c = 4
r = p1[0] + p2[0] + p3[0] + p4[0]
g = p1[0] + p2[0] + p3[0] + p4[0]
b = p1[0] + p2[0] + p3[0] + p4[0]
imagenFiltrada.putpixel((i,j),(r/c,g/c,b/c))
c = 0
return imagenFiltrada
m = I.open(argv[1])
gris = filtroGrisesPromedio(m)
bordesPorResta(gris,filtroPromedio(filtroPromedio(gris)))
pruido = float(argv[2])
generarRuido(m, pruido)from sys import argv
from PIL import Image as I
import random as R
def generarRuido(m, porcentaje):
imagenConRuido = m
p = m.load()
x,y = m.size
totalPixeles = x * y
pixelesMarcados = totalPixeles * porcentaje
pixelesMarcados = pixelesMarcados / 100
print pixelesMarcados
for i in range(int(pixelesMarcados)):
pxx = R.randrange(0,x)
pxy = R.randrange(0,y)
sop = R.uniform(0,1)
pixel = ''
if(sop > .5):
pixel = (255,255,255)
else:
pixel = (0,0,0)
imagenConRuido.putpixel((pxx,pxy),pixel)
imagenConRuido.show()
quitarRuidoPorVecindad(imagenConRuido)
def quitarRuidoPorVecindad(imagen):
m = []
imagen = imagen.convert('RGB')
x,y = imagen.size
pixeles = imagen.load()
c = 1
for i in range(x):
for j in range(y):
px = pixeles[i,j]
b = px[0]
if(b == 255 or b == 0):
m.append((i,j))
c = 0
try:
p1 = pixeles[i+1,j]
c += 1
except:
p1 = (0,0,0)
try:
p2 = pixeles[i-1,j]
c += 1
except:
p2 = (0,0,0)
try:
p3 = pixeles[i,j+1]
c += 1
except:
p3 = (0,0,0)
try:
p4 = pixeles[i,j-1]
c += 1
except:
p4 = (0,0,0)
if(c > 4):
c = 4
r = (p1[0] + p2[0] + p3[0] + p4[0])
g = (p1[0] + p2[0] + p3[0] + p4[0])
b = (p1[0] + p2[0] + p3[0] + p4[0])
pxnvo = (r/c,g/c,b/c)
imagen.putpixel((i,j),pxnvo)
c = 0
#SEGUNdA PASADA
fi = imagen.load()
for e in m:
c = 0
pixel = fi[e]
i,j = e
try:
p1 = fi[i+1,j]
c += 1
except:
p1 = (0,0,0)
try:
p2 = fi[i-1,j]
c += 1
except:
p2 = (0,0,0)
try:
p3 = fi[i,j+1]
c += 1
except:
p3 = (0,0,0)
try:
p4 = fi[i,j-1]
c += 1
except:
p4 = (0,0,0)
r = (p1[0] + p2[0] + p3[0] + p4[0])
g = (p1[0] + p2[0] + p3[0] + p4[0])
b = (p1[0] + p2[0] + p3[0] + p4[0])
pxnvo = (r/c,g/c,b/c)
imagen.putpixel((i,j),pxnvo)
c = 0
imagen.show()
return imagen
def bordesPorResta(im, imagen2):
x,y = im.size
m1 = im.load()
m2 = imagen2.load()
imb = I.new('RGB',(x,y))
for i in range(x):
for j in range(y):
r = m1[i,j][0] - m2[i,j][0]
g = m1[i,j][1] - m2[i,j][1]
b = m1[i,j][2] - m2[i,j][2]
pixel = (r,g,b)
imb.putpixel((i,j),pixel)
imb.show()
def filtroGrisesPromedio(imagen):
x, y = imagen.size
px = imagen.load()
imagenGrises = I.new('RGB',(x,y))
for i in range(x):
for j in range(y):
pixeles = px[i,j]
prom = sum(pixeles) / 3
imagenGrises.putpixel((i,j),(prom,prom,prom))
return imagenGrises
def filtroPromedio(imagen):
x,y = imagen.size
pixeles = imagen.load()
imagenFiltrada = I.new('RGB',(x,y))
c = 1
for i in range(x):
for j in range(y):
px = pixeles[i,j]
try:
p1 = pixeles[i+1,j]
c += 1
except:
p1 = (0,0,0)
try:
p2 = pixeles[i-1,j]
c += 1
except:
p2 = (0,0,0)
try:
p3 = pixeles[i,j+1]
c += 1
except:
p3 = (0,0,0)
try:
p4 = pixeles[i,j-1]
c += 1
except:
p4 = (0,0,0)
if(c > 4):
c = 4
r = p1[0] + p2[0] + p3[0] + p4[0]
g = p1[0] + p2[0] + p3[0] + p4[0]
b = p1[0] + p2[0] + p3[0] + p4[0]
imagenFiltrada.putpixel((i,j),(r/c,g/c,b/c))
c = 0
return imagenFiltrada
m = I.open(argv[1])
gris = filtroGrisesPromedio(m)
bordesPorResta(gris,filtroPromedio(filtroPromedio(gris)))
pruido = float(argv[2])
generarRuido(m, pruido)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment