AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-18 00:45:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed apply_background_color

Browse Source 
to remove partial transparency too if the specified background color is fully solid
This commit is contained in:
Lukas Bürgi 2025-01-27 12:14:09 +01:00
parent 37bbf63500
commit 06c6c57559
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
rembg/bg.py
View File

@ -175,9 +175,8 @@ def apply_background_color(img: PILImage, color: Tuple[int, int, int, int]) -> P
Returns:
PILImage: The modified image with the background color applied.
"""
r, g, b, a = color
colored_image = Image.new("RGBA", img.size, (r, g, b, a))
colored_image.paste(img, mask=img)
background = Image.new("RGBA", img.size, tuple(color))
colored_image = Image.alpha_composite(background, img)
return colored_image