Update app.py

app.py

@@ -23,6 +23,9 @@ def sobel(I):
     return Gx, Gy
 
 def medfilt2(G, d=3):
+    m, n = G.shape
+    temp = np.zeros([m+2*(d//2), n+2*(d//2)], np.float32)
+    salida = np.zeros([m, n], np.float32)
     temp[1:m+1, 1:n+1] = G
     for i in range(1, m):
         for j in range(1, n):
@@ -31,6 +34,10 @@ def medfilt2(G, d=3):
     return salida
 
 def orientacion(patron, w):
+    Gx, Gy = sobel(patron)
+    Gx = medfilt2(Gx)
+    Gy = medfilt2(Gy)
+    m, n = Gx.shape
     mOrientaciones = np.zeros([m//w, n//w], np.float32)
     for i in range(m//w):
         for j in range(n//w):