added colorReplacer

ColorReplacer.py

@@ -0,0 +1,31 @@
+import numpy as np
+from PIL import Image
+import cv2
+
+
+def hex_to_rgb(hex_color):
+    hex_color = hex_color.lstrip('#')
+    return tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4))
+def rgb_to_bgr(rgb_color):
+    return rgb_color[::-1]
+def recolor_selected_area(image_np, color_to_replace, replacement_color, tolerance=20):
+    # Convert hex colors to RGB and then to BGR
+    color_to_replace_rgb = hex_to_rgb(color_to_replace)
+    replacement_color_rgb = hex_to_rgb(replacement_color)
+    color_to_replace_bgr = rgb_to_bgr(color_to_replace_rgb)
+    replacement_color_bgr = rgb_to_bgr(replacement_color_rgb)
+    # Calculate the lower and upper bounds for color tolerance in BGR
+    lower_bound = np.array([max(c - tolerance, 0) for c in color_to_replace_bgr])
+    upper_bound = np.array([min(c + tolerance, 255) for c in color_to_replace_bgr])
+    # Convert the image to BGR format for OpenCV processing
+    image_bgr = cv2.cvtColor(image_np, cv2.COLOR_RGB2BGR)
+    # Create a mask for pixels within the tolerance range using cv2.inRange
+    mask = cv2.inRange(image_bgr, lower_bound, upper_bound)
+    # Replace color in the masked area
+    image_bgr[mask > 0] = replacement_color_bgr
+    # Convert back to RGB
+    result_image_np = cv2.cvtColor(image_bgr, cv2.COLOR_BGR2RGB)
+    # Save the result
+    result_image = Image.fromarray(result_image_np)
+    result_image.save('./result.jpg')
+    return result_image

recolorLinearColorTransfer.py

@@ -90,11 +90,14 @@ def create_target_image(hex_colors):
     
     return target_image
 
+
 def recolor(image_np, colors):
     colors = [color.replace('#', '') for color in colors]
     palette = create_target_image(colors)
     
-    # No need to convert to BGR and then read it again
+    # Only take the first three channels (RGB) of the image
+    image_np = image_np[:, :, :3]
+    
     target_img = image_np.astype(float)/256
     source_img = palette.astype(float)/256
     output_img = match_color(target_img, source_img)

server.py

@@ -15,6 +15,7 @@ import recolorPaletteBasedTransfer
 import recolorReinhardV2Algo
 import recolorLinearColorTransfer
 import matchCollection
+import ColorReplacer
 from typing import Optional
 
 app = FastAPI()
@@ -111,6 +112,9 @@ async def recolor(file: UploadFile = File(...), colors: str = Form(...),  model:
     elif method == "LinearColorTransfer":
         print('LinearColorTransfer generated')
         recolorLinearColorTransfer.recolor(image_np, colors)
+    elif method == "ColorReplacer":
+        print('ColorReplacer started')
+        ColorReplacer.recolor_selected_area(image_np, colors[0], colors[1])
 
     #mask image:
     if mask is not None:
@@ -129,7 +133,7 @@ async def recolor(file: UploadFile = File(...), colors: str = Form(...),  model:
         # Save the new image
         new_image = Image.fromarray(new_image_np)
         new_image.save('./result.jpg')
-        
+    
     img_file = open("./result.jpg", "rb")
     return StreamingResponse(img_file, media_type="image/jpeg")