Update texture_transfer.py

texture_transfer.py

@@ -1,71 +1,71 @@
-from PIL import Image, ImageOps
-import numpy as np
-from create_print_layover import create_hard_light_layover
-import cv2
-
-
-def create_layover(background_image, layer_image, opacity):
-    background_img_raw = Image.open(background_image)
-    background_img_raw = background_img_raw.convert("RGBA")
-    background_img = np.array(background_img_raw)
-    background_img_float = background_img.astype(float)
-    foreground_img_raw = Image.open(layer_image)
-    foreground_img_raw = foreground_img_raw.convert("RGBA")
-    foreground_img = np.array(foreground_img_raw)
-    foreground_img_float = foreground_img.astype(float)
-    blended_img_float = create_hard_light_layover(background_img_float, foreground_img_float, opacity)
-    blended_img = np.uint8(blended_img_float)
-    blended_img_raw = Image.fromarray(blended_img)
-    output_path = "lay_over_image.png"
-    blended_img_raw.save(output_path)
-    return output_path
-
-
-def create_image_tile(input_patch, x_dim, y_dim):
-    input_image = Image.open(input_patch)
-    input_image = input_image.convert("RGB")
-    width, height = input_image.size
-    output_image = Image.new("RGB", (x_dim, y_dim))
-    for y in range(0, y_dim, height):
-        for x in range(0, x_dim, width):
-            region_height = min(height, y_dim - y)
-            region_width = min(width, x_dim - x)
-            region = input_image.crop((0, 0, region_width, region_height))
-            output_image.paste(region, (x, y))
-    output_image.save('tiled_image.png')
-
-
-def create_image_cutout(texture_transfer_image, actual_mask):
-    image = Image.open(texture_transfer_image)
-    mask = Image.open(actual_mask).convert('L')
-    if mask.size != image.size:
-        image = image.resize(mask.size, Image.Resampling.NEAREST)
-    image_np = np.array(image)
-    mask_np = np.array(mask)
-    binary_mask = (mask_np > 127).astype(np.uint8) * 255
-    masked_image_np = image_np * np.expand_dims(binary_mask, axis=-1) // 255
-    masked_image = Image.fromarray(masked_image_np)
-    masked_image.save('cut_out_image.png')
-
-
-def paste_image(base_image_path, cutout_image_path, mask_path):
-    background = Image.open(base_image_path).convert("RGB")
-    cutout = Image.open(cutout_image_path)
-    mask = Image.open(mask_path).convert("L")
-    if cutout.mode == 'RGBA':
-        cutout_rgb = cutout.convert("RGB")
-        cutout_alpha = cutout.split()[-1]
-    else:
-        cutout_rgb = cutout
-        cutout_alpha = mask
-    cutout_rgb = cutout_rgb.resize(background.size, Image.Resampling.NEAREST)
-    cutout_alpha = cutout_alpha.resize(background.size, Image.Resampling.NEAREST)
-    cutout_alpha_np = np.array(cutout_alpha)
-    binary_mask = (cutout_alpha_np > 128).astype(np.uint8) * 255
-    cutout_rgb_np = np.array(cutout_rgb)
-    background_np = np.array(background)
-    cutout_masked = cutout_rgb_np * np.expand_dims(binary_mask, axis=-1) // 255
-    background_masked = background_np * np.expand_dims(255 - binary_mask, axis=-1) // 255
-    result_np = cutout_masked + background_masked
-    result = Image.fromarray(result_np.astype(np.uint8))
-    result.save('result.png')
+from PIL import Image, ImageOps
+import numpy as np
+from create_print_layover import create_hard_light_layover
+import cv2
+
+
+def create_layover(background_image, layer_image, opacity):
+    # background_img_raw = Image.open(background_image)
+    background_img_raw = background_image.convert("RGBA")
+    background_img = np.array(background_img_raw)
+    background_img_float = background_img.astype(float)
+    # foreground_img_raw = Image.open(layer_image)
+    foreground_img_raw = layer_image.convert("RGBA")
+    foreground_img = np.array(foreground_img_raw)
+    foreground_img_float = foreground_img.astype(float)
+    blended_img_float = create_hard_light_layover(background_img_float, foreground_img_float, opacity)
+    blended_img = np.uint8(blended_img_float)
+    blended_img_raw = Image.fromarray(blended_img)
+    output_path = "lay_over_image.png"
+    blended_img_raw.save(output_path)
+    return output_path
+
+
+def create_image_tile(input_patch, x_dim, y_dim):
+    input_image = Image.open(input_patch)
+    input_image = input_image.convert("RGB")
+    width, height = input_image.size
+    output_image = Image.new("RGB", (x_dim, y_dim))
+    for y in range(0, y_dim, height):
+        for x in range(0, x_dim, width):
+            region_height = min(height, y_dim - y)
+            region_width = min(width, x_dim - x)
+            region = input_image.crop((0, 0, region_width, region_height))
+            output_image.paste(region, (x, y))
+    output_image.save('tiled_image.png')
+
+
+def create_image_cutout(texture_transfer_image, actual_mask):
+    image = Image.open(texture_transfer_image)
+    mask = Image.open(actual_mask).convert('L')
+    if mask.size != image.size:
+        image = image.resize(mask.size, Image.Resampling.NEAREST)
+    image_np = np.array(image)
+    mask_np = np.array(mask)
+    binary_mask = (mask_np > 127).astype(np.uint8) * 255
+    masked_image_np = image_np * np.expand_dims(binary_mask, axis=-1) // 255
+    masked_image = Image.fromarray(masked_image_np)
+    masked_image.save('cut_out_image.png')
+
+
+def paste_image(base_image_path, cutout_image_path, mask_path):
+    background = Image.open(base_image_path).convert("RGB")
+    cutout = Image.open(cutout_image_path)
+    mask = Image.open(mask_path).convert("L")
+    if cutout.mode == 'RGBA':
+        cutout_rgb = cutout.convert("RGB")
+        cutout_alpha = cutout.split()[-1]
+    else:
+        cutout_rgb = cutout
+        cutout_alpha = mask
+    cutout_rgb = cutout_rgb.resize(background.size, Image.Resampling.NEAREST)
+    cutout_alpha = cutout_alpha.resize(background.size, Image.Resampling.NEAREST)
+    cutout_alpha_np = np.array(cutout_alpha)
+    binary_mask = (cutout_alpha_np > 128).astype(np.uint8) * 255
+    cutout_rgb_np = np.array(cutout_rgb)
+    background_np = np.array(background)
+    cutout_masked = cutout_rgb_np * np.expand_dims(binary_mask, axis=-1) // 255
+    background_masked = background_np * np.expand_dims(255 - binary_mask, axis=-1) // 255
+    result_np = cutout_masked + background_masked
+    result = Image.fromarray(result_np.astype(np.uint8))
+    result.save('result.png')