| from typing import Tuple | |
| import PIL | |
| from PIL.Image import Image as PILImage | |
| from PIL import ImageOps | |
| def get_background_dominant_color(img: PILImage, mask: PILImage) -> tuple: | |
| negative_img = img.copy() | |
| negative_mask = PIL.ImageOps.invert(mask).convert("L") | |
| negative_img.putalpha(negative_mask) | |
| negative_img = negative_img.resize((1, 1)) | |
| r, g, b, a = negative_img.getpixel((0, 0)) | |
| return r, g, b, 255 | |
| def apply_background_color(img: PILImage, color: Tuple[int, int, int, int]) -> PILImage: | |
| r, g, b, a = color | |
| colored_image = PIL.Image.new("RGBA", img.size, (r, g, b, a)) | |
| colored_image.paste(img, mask=img) | |
| return colored_image | |
| def make_flatten_background(img_original, img_cutted, mask): | |
| img_original = PIL.Image.open(img_original) | |
| img_cutted = PIL.Image.open(img_cutted) | |
| mask = PIL.Image.open(mask) | |
| color = get_background_dominant_color(img_original, mask) | |
| return apply_background_color(img_cutted, color) | |