Spaces:
Sleeping
Sleeping
File size: 1,022 Bytes
da3fe02 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
from __future__ import annotations
from pathlib import Path
from typing import Optional
from PIL import Image
def load_image(
path: Path,
upscale: int,
alpha_background: str,
) -> Image.Image:
image = Image.open(path)
image = _apply_alpha(image, alpha_background)
if upscale and upscale > 1:
image = image.resize(
(image.width * upscale, image.height * upscale),
resample=Image.BICUBIC,
)
return image
def _apply_alpha(image: Image.Image, alpha_background: str) -> Image.Image:
if image.mode in ("RGBA", "LA") or (image.mode == "P" and "transparency" in image.info):
if alpha_background == "none":
return image.convert("RGBA")
color = (255, 255, 255, 255) if alpha_background == "white" else (0, 0, 0, 255)
background = Image.new("RGBA", image.size, color)
foreground = image.convert("RGBA")
return Image.alpha_composite(background, foreground).convert("RGB")
return image.convert("RGB")
|