Create utils.py

utils.py

@@ -0,0 +1,10 @@
+import cv2
+import numpy as np
+from PIL import Image
+
+def generate_canny(image: Image.Image, low_threshold=50, high_threshold=150) -> Image.Image:
+    image = image.resize((1024, 1024))
+    img_np = np.array(image)
+    edges = cv2.Canny(img_np, low_threshold, high_threshold)
+    edges_rgb = cv2.cvtColor(edges, cv2.COLOR_GRAY2RGB)
+    return Image.fromarray(edges_rgb)