doing color transfer before changing garment

app.py

@@ -154,6 +154,9 @@ def start_tryon(human_img_dict,garm_img,garment_des, background_img, is_checked,
     #human_img_orig = human_img_dict["background"].convert("RGB")   # ImageEditor
     human_img_orig = human_img_dict.convert("RGB")     # Image
 
+    if background_img:
+        human_img_orig = BackgroundProcessor.color_transfer(human_img_orig, background_img)
+
     """
     # Derive HEIGHT & WIDTH such that width is not more than 1000. This will cater to both Shein images (4160x6240) of 3:4 AR and model standard images ( 768x1024 ) of 2:3 AR
     WIDTH, HEIGHT = human_img_orig.size

src/background_processor.py

@@ -196,7 +196,7 @@ class BackgroundProcessor:
         #background_image_pil = background_image_pil.resize((width, height)) 
         
         # Do color transfer of background to foreground to adjust lighting condition
-        foreground_img_pil = cls.color_transfer(foreground_img_pil, background_image_pil)
+        #foreground_img_pil = cls.color_transfer(foreground_img_pil, background_image_pil)
 
         foreground_binary = ImageFormatConvertor.pil_image_to_binary_data(foreground_img_pil)
         background_binary = ImageFormatConvertor.pil_image_to_binary_data(background_image_pil)