Update sketch_helper.py

sketch_helper.py

@@ -54,4 +54,22 @@ def create_binary_matrix(img_arr, target_color):
     cv2.imwrite(binary_file_name, binary_matrix * 255)
     
     #binary_matrix = torch.from_numpy(binary_matrix).unsqueeze(0).unsqueeze(0)
-    return binary_file_name
+    return binary_file_name
+
+def color_distance(color1, color2):
+    return sum((a - b) ** 2 for a, b in zip(color1, color2)) ** 0.5
+
+def replace_color(image, old_color, new_color):
+    pixels = image.load()
+    width, height = image.size
+    for x in range(width):
+        for y in range(height):
+            if pixels[x, y] == old_color:
+                pixels[x, y] = new_color
+
+def opaque_color_on_white(color, a):
+    r, g, b = color
+    opaque_red = int((1 - a) * 255 + a * r)
+    opaque_green = int((1 - a) * 255 + a * g)
+    opaque_blue = int((1 - a) * 255 + a * b)
+    return (opaque_red, opaque_green, opaque_blue)