Update filters.py

filters.py

@@ -38,4 +38,13 @@ def edge_enhance(image, intensity: float = 1.5):
         [-1 * intensity,  9 * intensity, -1 * intensity],
         [-1 * intensity, -1 * intensity, -1 * intensity]
     ])
-    return cv2.filter2D(image, -1, kernel)
+    return cv2.filter2D(image, -1, kernel)
+
+@registry.register("Canny Edge", defaults={'lower_threshold': 100, 'upper_threshold': 200, 'convert_to_gray': True}, min_vals={'lower_threshold': 0, 'upper_threshold': 255}, max_vals={'lower_threshold': 255, 'upper_threshold': 255}, step_vals={'lower_threshold': 1, 'upper_threshold': 1})
+def canny_edge(image, lower_threshold=100, upper_threshold=200, convert_to_gray=True):
+    if convert_to_gray and len(image.shape) == 3:
+        gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    else:
+        gray = image
+    edges = cv2.Canny(gray, lower_threshold, upper_threshold)
+    return edges