Update utils/image_utils.py

utils/image_utils.py

@@ -5,7 +5,7 @@ from rembg import remove
 
 def background_removal(input_image_path):
     """
-    指定された画像から背景を除去し、透明部分のみを白背景に置き換えた画像を返す関数
+    指定された画像から背景を除去し、透明部分を白背景にブレンドして返す関数
     """
     try:
         input_image = Image.open(input_image_path)
@@ -16,15 +16,13 @@ def background_removal(input_image_path):
     # 背景除去処理
     rgba_image = remove(input_image).convert("RGBA")
     rgba_np = np.array(rgba_image)
-    alpha_channel = rgba_np[:, :, 3]
+    alpha_channel = rgba_np[:, :, 3] / 255.0  # アルファを0~1の範囲に正規化
 
-    # 元画像と白背景のマスク作成
-    foreground = rgba_np[:, :, :3]  # RGB部分のみ取得
-    background = np.ones_like(foreground, dtype=np.uint8) * 255  # 白背景
-    
-    # 透過部分のみを白で塗りつぶす
-    # アルファマスクを使って背景（白）を透過部分に合成
-    result = np.where(alpha_channel[:, :, None] == 0, background, foreground)
+    # 白背景の生成
+    background = np.ones_like(rgba_np[:, :, :3], dtype=np.uint8) * 255  # 白背景
+
+    # アルファブレンドを行い、透明部分を白と合成
+    result = (foreground * alpha_channel[:, :, None] + background * (1 - alpha_channel[:, :, None])).astype(np.uint8)
     
     return Image.fromarray(result)