eienmojiki commited on
Commit
b8f7d24
·
verified ·
1 Parent(s): 5e61196

Update filters.py

Browse files
Files changed (1) hide show
  1. 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=15)
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=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)
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)