-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathFiltros_Convolução.py
111 lines (104 loc) · 3.06 KB
/
Filtros_Convolução.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
import sys
import cv
import os
class Filtros:
def __init__(self,imagem_cinza):
self.imagem_cinza = imagem_cinza
versao = 0
cv.NamedWindow("Foto Cinza", 1)
cv.NamedWindow("Foto Filtrada", 1)
cv.ShowImage("Foto Cinza",imagem_cinza)
cv.CreateTrackbar("Filtros", "Foto Cinza", 0, 7, self.atualiza_matriz)
self.atualiza_matriz(versao)
def convolui_com_uma_matriz(self,matriz):
imagem_resultado = cv.CreateImage( cv.GetSize(imagem), 8, 1 )
print("Multiplicando ...\n")
for x in range(imagem.height-1):
for y in range(imagem.width-1):
soma,magnitude = 0 , 0
for i in range(-1,2):
for j in range(-1,2):
soma+=(self.imagem_cinza[x+i,y+j]*matriz[i+1][j+1])
magnitude = abs(soma)
if magnitude > 255 : magnitude = 255
imagem_resultado[x,y] = magnitude
print("Pronto !")
cv.ShowImage("Foto Filtrada",imagem_resultado)
def convolui_com_duas_matrizes(self,matriz_X,matriz_Y):
imagem_resultado = cv.CreateImage( cv.GetSize(imagem), 8, 1 )
print("Multiplicando ...\n")
for x in range(imagem.height-1):
for y in range(imagem.width-1):
soma_X,soma_Y,magnitude = 0 , 0 , 0
for i in range(-1,2):
for j in range(-1,2):
soma_X+=(self.imagem_cinza[x+i,y+j]*matriz_X[i+1][j+1])
soma_Y+=(self.imagem_cinza[x+i,y+j]*matriz_Y[i+1][j+1])
magnitude = abs(soma_X) + abs(soma_Y)
if magnitude > 255 : magnitude = 255
imagem_resultado[x,y] = magnitude
print("Pronto !")
cv.ShowImage("Foto Filtrada",imagem_resultado)
def atualiza_matriz(self,versao):
os.system('clear')
if (versao == 0 ):
print("Laplace [1]")
matriz = [0,-1,0],\
[-1,4,-1],\
[0,-1,0]
self.convolui_com_uma_matriz(matriz)
if (versao == 1 ):
print("Laplace [2]")
matriz = [-1,-1,-1],\
[-1,8,-1],\
[-1,-1,-1]
self.convolui_com_uma_matriz(matriz)
if (versao == 2 ):
print("Sobel [X]")
matriz = [-1,-2,-1],\
[0,0,0],\
[1,2,1]
self.convolui_com_uma_matriz(matriz)
if (versao == 3 ):
print("Sobel [Y]")
matriz = [-1,0,1],\
[-2,0,2],\
[-1,0,1]
self.convolui_com_uma_matriz(matriz)
if (versao == 4 ):
print("Sobel [X][Y]")
matriz_X = [-1,0,1],\
[-2,0,2],\
[-1,0,1]
matriz_Y = [-1,-2,-1],\
[0,0,0],\
[1,2,1]
self.convolui_com_duas_matrizes(matriz_X,matriz_Y)
if (versao == 5 ):
print("Prewitt [X]")
matriz = [-1,-1,-1],\
[0,0,0],\
[1,1,1]
self.convolui_com_uma_matriz(matriz)
if (versao == 6 ):
print("Prewitt [Y]")
matriz = [-1,0,1],\
[-1,0,1],\
[-1,0,1]
self.convolui_com_uma_matriz(matriz)
if (versao == 7 ):
print("Prewitt [X][Y]")
matriz_X = [-1,0,1],\
[-1,0,1],\
[-1,0,1]
matriz_Y = [-1,-1,-1],\
[0,0,0],\
[1,1,1]
self.convolui_com_duas_matrizes(matriz_X,matriz_Y)
if __name__ == "__main__":
imagem = cv.LoadImage('foto.jpg')
imagem_cinza = cv.CreateImage( cv.GetSize(imagem), 8, 1 )
cv.CvtColor(imagem,imagem_cinza, cv.CV_BGR2GRAY )
filtros = Filtros(imagem_cinza)
cv.WaitKey(0)
print("Elisson Michael [UENF] : Processamento de Imagens")