new modes added, fixes in old methods and improvements in the server

recolorLinearColorTransfer.py

@@ -5,35 +5,36 @@
 #based on this: https://github.com/ProGamerGov/Neural-Tools
 
 import numpy as np
-import argparse
+# import argparse
 import imageio
 from skimage import io,transform,img_as_float
 from skimage.io import imread,imsave
 from PIL import Image
-from numpy import eye 
+from numpy import eye
+import cv2 
 
-parser = argparse.ArgumentParser()
-parser.add_argument('-t', '--target_image', type=str, help="The image you are transfering color to. Ex: target.png", required=True)
-parser.add_argument('-s', '--source_image', type=str, help="The image you are transfering color from. Ex: source.png", required=True)
-parser.add_argument('-o', '--output_image', default='output.png', help="The name of your output image. Ex: output.png", type=str)
-parser.add_argument('-m', '--mode', default='pca', help="The color transfer mode. Options are pca, chol, or sym.", type=str)
-parser.add_argument('-e', '--eps', default='1e-5', help="Your epsilon value in scientific notation or normal notation. Ex: 1e-5 or 0.00001", type=float)
-parser.parse_args()
-args = parser.parse_args()
+# parser = argparse.ArgumentParser()
+# parser.add_argument('-t', '--target_image', type=str, help="The image you are transfering color to. Ex: target.png", required=True)
+# parser.add_argument('-s', '--source_image', type=str, help="The image you are transfering color from. Ex: source.png", required=True)
+# parser.add_argument('-o', '--output_image', default='output.png', help="The name of your output image. Ex: output.png", type=str)
+# parser.add_argument('-m', '--mode', default='pca', help="The color transfer mode. Options are pca, chol, or sym.", type=str)
+# parser.add_argument('-e', '--eps', default='1e-5', help="Your epsilon value in scientific notation or normal notation. Ex: 1e-5 or 0.00001", type=float)
+# parser.parse_args()
+# args = parser.parse_args()
 
 
 Image.MAX_IMAGE_PIXELS = 1000000000 # Support gigapixel images
 
 
-def main():   
+# def main():   
 
-    target_img = imageio.v2.imread(args.target_image, pilmode="RGB").astype(float)/256
-    source_img = imageio.v2.imread(args.source_image, pilmode="RGB").astype(float)/256
+#     target_img = imageio.v2.imread(args.target_image, pilmode="RGB").astype(float)/256
+#     source_img = imageio.v2.imread(args.source_image, pilmode="RGB").astype(float)/256
     
-    output_img = match_color(target_img, source_img, mode=args.mode, eps=args.eps)
-    output_img = (output_img * 255).astype(np.uint8)
-    imsave(args.output_image, output_img)
-    # imsave(args.output_image, output_img)
+#     output_img = match_color(target_img, source_img, mode=args.mode, eps=args.eps)
+#     output_img = (output_img * 255).astype(np.uint8)
+#     imsave(args.output_image, output_img)
+#     # imsave(args.output_image, output_img)
 
 
 def match_color(target_img, source_img, mode='pca', eps=1e-5):
@@ -75,5 +76,28 @@ def match_color(target_img, source_img, mode='pca', eps=1e-5):
     return matched_img
 
 
-if __name__ == "__main__":
-    main()
+# if __name__ == "__main__":
+#     main()
+
+def create_target_image(hex_colors):
+    # Convert hex colors to RGB
+    rgb_colors = [tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4)) for hex_color in hex_colors]
+    
+    # Create an image with the RGB colors
+    target_image = np.zeros((100, len(rgb_colors), 3), dtype=np.uint8)
+    for i, rgb_color in enumerate(rgb_colors):
+        target_image[:, i] = rgb_color
+    
+    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
+    target_img = image_np.astype(float)/256
+    source_img = palette.astype(float)/256
+    output_img = match_color(target_img, source_img)
+    output_img = (output_img * 255).astype(np.uint8)
+    imsave('result.jpg', output_img)
+    return output_img

recolorPaletteBasedTransfer.py

@@ -19,7 +19,7 @@ def extract_palette(image, n_colors=5):
     pixels = image.reshape(-1, 3)
     
     # Apply KMeans clustering to find the most dominant colors
-    kmeans = KMeans(n_clusters=n_colors)
+    kmeans = KMeans(n_clusters=n_colors, n_init='auto')
     kmeans.fit(pixels)
     
     # Get the RGB values of the clusters' centers
@@ -70,8 +70,13 @@ def palette_based_color_transfer(source_img, target_palette, n_colors=5):
     # Extract the source palette
     source_palette = extract_palette(source_img, n_colors)
     
+    # Ensure target_palette has the same number of colors as source_palette
+    while len(target_palette) < n_colors:
+        target_palette += target_palette
+
     # Perform color mapping
-    recolored_img = map_colors(source_img, source_palette, target_palette)
+    
+    recolored_img = map_colors(source_img, source_palette, target_palette[:n_colors])
     
     return recolored_img
 

recolorReinhardV2Algo.py

@@ -37,12 +37,45 @@ def color_transfer(source, target):
     
     return result
 
-# Load source and target images
-source_img = cv2.imread("estampa.jpg")
-target_img = cv2.imread("cor.jpg")
 
-# Perform color transfer
-transferred_img = color_transfer(source_img, target_img)
 
-# Return the transferred image for visualization
-cv2.imwrite("result.jpg", transferred_img)
+def create_target_image(hex_colors):
+    # Convert hex colors to RGB
+    rgb_colors = [tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4)) for hex_color in hex_colors]
+    
+    # Create an image with the RGB colors
+    target_image = np.zeros((100, len(rgb_colors), 3), dtype=np.uint8)
+    for i, rgb_color in enumerate(rgb_colors):
+        target_image[:, i] = rgb_color
+    
+    return target_image
+
+
+# # Load source and target images
+# source_img = cv2.imread("estampa.jpg")
+# target_img = cv2.imread("cor.jpg")
+
+
+# hex_colors = ['ff0000', '00ff00', '0000ff']  # Example hex colors
+# target_img = create_target_image(hex_colors)
+
+# image_bytes = io.BytesIO(source_img)
+# image_np = np.array(image)
+
+# # Perform color transfer
+# transferred_img = color_transfer(source_img, target_img)
+
+# # Return the transferred image for visualization
+# cv2.imwrite("result.jpg", transferred_img)
+
+def recolor(image_np, colors):
+    colors = [color.replace('#', '') for color in colors]
+    
+    target_img = create_target_image(colors)
+    source_bgr = cv2.cvtColor(image_np, cv2.COLOR_RGB2BGR)
+    transferred_img = color_transfer(source_bgr, target_img)
+    cv2.imwrite("result.jpg", transferred_img)
+    return transferred_img
+
+
+

server.py

@@ -12,6 +12,8 @@ import recolorOTAlgo
 import recolorTransferAlgo
 import recolorLumaConverterAlgo
 import recolorPaletteBasedTransfer
+import recolorReinhardV2Algo
+import recolorLinearColorTransfer
 
 app = FastAPI()
 
@@ -82,20 +84,31 @@ async def recolor(file: UploadFile = File(...), colors: str = Form(...),  model:
     image_np = np.array(image)
 
     if method == "CCA":
+        print('CCA generated')
         #Characteristic Color Analysis
         recolorReinhardAlgo.recolor(image_np, colors)
     elif method == "OTA":
+        print('OTA generated')
         #Optimal Transport Algorithm transfer
         recolorOTAlgo.recolor(image_np, colors)
     elif method =="KMEANS":
+        print('KMEANS generated')
         #K-means clustering transfer
         recolorTransferAlgo.recolor(image_np, colors)
     elif method == "LUMA":
+        print('Luma generated')
         #Luma converter transfer
         recolorLumaConverterAlgo.remap_image_colors(image_np, colors)
     elif method == "palette":
         #palette transfer
+        print('palette generated')
         recolorPaletteBasedTransfer.recolor(image_np, colors)
+    elif method == "Reinhardv2":
+        print('Reinhardv2 generated')
+        recolorReinhardV2Algo.recolor(image_np, colors)
+    elif method == "LinearColorTransfer":
+        print('LinearColorTransfer generated')
+        recolorLinearColorTransfer.recolor(image_np, colors)
     img_file = open("./result.jpg", "rb")
     return StreamingResponse(img_file, media_type="image/jpeg")
 @app.get("/test/")

todo.md

@@ -1,10 +1,10 @@
 * add more two funcoes de troca de cor
 *add invert colors
-*check file type
+*check file type (maybe not needed?)
 *mask
 
 **added:
-recolor palette based transfer, recolor reinhard v2 and recolor linear transfer. => need to add to the backend.
+recolor linear transfer. => need to add to the backend.
 
 `
 source ./env/bin/activate
@@ -12,9 +12,5 @@ source ./env/bin/activate
 #how to run:
 #source env/bin/activate     
 #uvicorn server:app --reload
+#docker run -p 4201:7860 farm-recolor    
 `
-
-
-ERRORS TODO:
-PALETTE ALGO IS WORKING BUT GIVING WARNINGS AND ERRORS.
-DOMINANT COLOR IS BROKEN