X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/f0d17e971a75a2d4d86b64877d6de51da3fad2a9..442f568116a71d79586ee44bb31ae1b96d233a91:/filters.py?ds=sidebyside

diff --git a/filters.py b/filters.py
index c2648ea..ce955a4 100644
--- a/filters.py
+++ b/filters.py
@@ -1,17 +1,17 @@
 from PIL import Image
 
-def edge_detection(image):
+def edge_detection(image, size):
     image_l = image.load()
     new_image = Image.new('L', image.size)
     new_image_l = new_image.load()
     
-    for x in xrange(2, image.size[0] - 2):
-        for y in xrange(2, image.size[1] - 2):
+    for x in xrange(size, image.size[0] - size):
+        for y in xrange(size, image.size[1] - size):
             pix = (sum([sum([
                 image_l[a, b] 
-                    for b in range(y - 2, y + 3)]) 
-                    for a in range(x - 2, x + 3)])
-                - (25 * image_l[x, y]))
+                    for b in range(y - size, y + size + 1)]) 
+                    for a in range(x - size, x + size + 1)])
+                - (((2 * size + 1) ** 2)  * image_l[x, y]))
             if pix > 255:
                 pix = 255
             if pix < 0: