match color function

src/background_processor.py

@@ -232,8 +232,10 @@ class BackgroundProcessor:
         logging.error(f"failed to use remove bg. Status: {remove_bg_request.status_code}. Resp: {remove_bg_request.content}")
         return None
     
+    
     @classmethod
     def color_transfer(cls, source_pil: Image, target_pil: Image) -> Image:
+        # NOT IN USE as output color was not good
         source = ImageFormatConvertor.pil_to_cv2(source_pil)
         # Resize background image
         width, height = source_pil.width, source_pil.height
@@ -263,6 +265,7 @@ class BackgroundProcessor:
 
     @classmethod
     def intensity_transfer(cls, source_pil: Image, target_pil: Image) -> Image:
+
         """
         Transfers the intensity distribution from the target image to the source image.
 
@@ -306,4 +309,31 @@ class BackgroundProcessor:
         # return cv2.cvtColor(result_lab, cv2.COLOR_LAB2BGR)
         res = cv2.cvtColor(result_lab, cv2.COLOR_LAB2BGR) 
         res_pil = ImageFormatConvertor.cv2_to_pil(res)
-        return res_pil
+        return res_pil
+    
+    @classmethod
+    def match_color(cls, source_pil: Image, target_pil: Image):
+        source = ImageFormatConvertor.pil_to_cv2(source_pil)
+        # Resize background image
+        width, height = source_pil.width, source_pil.height
+        target_pil = target_pil.convert("RGB")
+        target_pil = target_pil.resize((width, height))
+
+        target = ImageFormatConvertor.pil_to_cv2(target_pil)
+        
+        matched_foreground = cv2.cvtColor(source, cv2.COLOR_BGR2LAB)
+        matched_background = cv2.cvtColor(target, cv2.COLOR_BGR2LAB)
+        
+        # Match the histograms
+        for i in range(3):
+            matched_foreground[:, :, i] = cv2.equalizeHist(matched_foreground[:, :, i])
+            matched_background[:, :, i] = cv2.equalizeHist(matched_background[:, :, i])
+        
+        matched_foreground = cv2.cvtColor(matched_foreground, cv2.COLOR_LAB2BGR)
+        matched_background = cv2.cvtColor(matched_background, cv2.COLOR_LAB2BGR)
+
+        matched_foreground_pil = ImageFormatConvertor.cv2_to_pil(matched_foreground)
+        matched_background_pil = ImageFormatConvertor.cv2_to_pil(matched_background)
+
+        return matched_foreground_pil, matched_background_pil
+