Spaces:
Running
Running
Update filters.py
Browse files- filters.py +3 -7
filters.py
CHANGED
@@ -10,7 +10,7 @@ def original(image):
|
|
10 |
def grayscale(image):
|
11 |
return cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
|
12 |
|
13 |
-
@registry.register("Gaussian Blur", kernel_size=
|
14 |
def gaussian_blur(image, kernel_size: int = 15):
|
15 |
return cv2.GaussianBlur(image, (kernel_size, kernel_size), 0)
|
16 |
|
@@ -31,15 +31,11 @@ def sepia(image):
|
|
31 |
])
|
32 |
return cv2.transform(image, kernel)
|
33 |
|
34 |
-
@registry.register("Edge Enhance", intensity
|
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)
|
42 |
-
|
43 |
-
@registry.register("Adjust Contrast", alpha=1.5)
|
44 |
-
def adjust_contrast(image, alpha: float = 1.5):
|
45 |
-
return cv2.convertScaleAbs(image, alpha=alpha, beta=0)
|
|
|
10 |
def grayscale(image):
|
11 |
return cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
|
12 |
|
13 |
+
@registry.register("Gaussian Blur", defaults={'kernel_size': 15}, min_vals={'kernel_size': 3}, max_vals={'kernel_size': 31}, step_vals={'kernel_size': 2})
|
14 |
def gaussian_blur(image, kernel_size: int = 15):
|
15 |
return cv2.GaussianBlur(image, (kernel_size, kernel_size), 0)
|
16 |
|
|
|
31 |
])
|
32 |
return cv2.transform(image, kernel)
|
33 |
|
34 |
+
@registry.register("Edge Enhance", defaults={'intensity': 1.5}, min_vals={'intensity': 0.5}, max_vals={'intensity': 5.0}, step_vals={'intensity': 0.1})
|
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)
|
|
|
|
|
|
|
|