Files changed (1) hide show
  1. filters.py +41 -0
filters.py ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import cv2
2
+ import numpy as np
3
+ from registry import registry
4
+
5
+ @registry.register("Original")
6
+ def original(image):
7
+ return image
8
+
9
+ @registry.register("Grayscale")
10
+ def grayscale(image):
11
+ return cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
12
+
13
+ @registry.register("Gaussian Blur", kernel_size=15)
14
+ def gaussian_blur(image, kernel_size: int = 15):
15
+ return cv2.GaussianBlur(image, (kernel_size, kernel_size), 0)
16
+
17
+ @registry.register("Pencil Sketch")
18
+ def pencil_sketch(image):
19
+ gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
20
+ inverted = cv2.bitwise_not(gray)
21
+ blurred = cv2.GaussianBlur(inverted, (21, 21), 0)
22
+ inverted_blurred = cv2.bitwise_not(blurred)
23
+ return cv2.divide(gray, inverted_blurred, scale=256.0)
24
+
25
+ @registry.register("Sepia")
26
+ def sepia(image):
27
+ kernel = np.array([
28
+ [0.393, 0.769, 0.189],
29
+ [0.349, 0.686, 0.168],
30
+ [0.272, 0.534, 0.131]
31
+ ])
32
+ return cv2.transform(image, kernel)
33
+
34
+ @registry.register("Edge Enhance", intensity=1.5)
35
+ def edge_enhance(image, intensity: float = 1.5):
36
+ kernel = np.array([
37
+ [-1 * intensity, -1 * intensity, -1 * intensity],
38
+ [-1 * intensity, 9 * intensity, -1 * intensity],
39
+ [-1 * intensity, -1 * intensity, -1 * intensity]
40
+ ])
41
+ return cv2.filter2D(image, -1, kernel)