Kavindu99 commited on
Commit
ddb49c7
·
1 Parent(s): 43cb5ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -5
app.py CHANGED
@@ -5,6 +5,14 @@ from PIL import Image
5
 
6
 
7
  def convolve(img, kernel):
 
 
 
 
 
 
 
 
8
  sobel_verticle = np.array([[-1,0,1],
9
  [-1,0,1],
10
  [-1,0,1]])
@@ -25,14 +33,27 @@ def convolve(img, kernel):
25
  weighted_average = np.array([[1/16,2/16, 1/16],
26
  [2/16,4/16, 2/16],
27
  [1/16,2/16, 1/16]])
 
 
 
 
 
 
 
28
 
29
- out_img = cv2.filter2D(src=img, ddepth =-1, kernel=kernel)
30
- out_img = cv2.cvtColor(out_img, cv2.COLOR_BGR2RGB)
31
- out_img = cv2.resize(out_img, (226,226))
32
- return Image.fromarray(out_img)
33
 
 
 
 
 
 
 
 
 
34
 
35
- def app(img, filter):
36
  if filter =="sobel_verticle":
37
  return convolve(img, sobel_verticle)
38
  elif filter =="sobel_horizontal":
 
5
 
6
 
7
  def convolve(img, kernel):
8
+
9
+ out_img = cv2.filter2D(src=img, ddepth =-1, kernel=kernel)
10
+ out_img = cv2.cvtColor(out_img, cv2.COLOR_BGR2RGB)
11
+ out_img = cv2.resize(out_img, (226,226))
12
+ return Image.fromarray(out_img)
13
+
14
+
15
+ def app(img, filter):
16
  sobel_verticle = np.array([[-1,0,1],
17
  [-1,0,1],
18
  [-1,0,1]])
 
33
  weighted_average = np.array([[1/16,2/16, 1/16],
34
  [2/16,4/16, 2/16],
35
  [1/16,2/16, 1/16]])
36
+ sobel_verticle = np.array([[-1,0,1],
37
+ [-1,0,1],
38
+ [-1,0,1]])
39
+
40
+ sobel_horizontal = np.array([[-1,-1,11],
41
+ [ 0, 0, 0],
42
+ [ 1, 1, 1]])
43
 
44
+ laplacian = np.array([[ 0,-1, 0],
45
+ [-1, 5,-1],
46
+ [ 0,-1, 0]])
 
47
 
48
+ average = np.array([[1/9,1/9, 1/9],
49
+ [1/9,1/9, 1/9],
50
+ [1/9,1/9, 1/9]])
51
+
52
+
53
+ weighted_average = np.array([[1/16,2/16, 1/16],
54
+ [2/16,4/16, 2/16],
55
+ [1/16,2/16, 1/16]])
56
 
 
57
  if filter =="sobel_verticle":
58
  return convolve(img, sobel_verticle)
59
  elif filter =="sobel_horizontal":