Function expand

detikzify/util/image.py:30–33

Expand image to a square of size {size}. Optionally trims borders first.

(image, size, do_trim=False, bg="white")

    return image.crop(bbox) if (bbox:=diff.getbbox()) else image

def expand(image, size, do_trim=False, bg="white"):
    """Expand image to a square of size {size}. Optionally trims borders first."""
    image = trim(image, bg=bg) if do_trim else image
    return ImageOps.pad(image, (size, size), color=bg, method=Image.Resampling.LANCZOS)

#  based on transformers/image_utils.py (added support for rgba images)
def load(image: Image.Image | str | bytes, bg="white", timeout=None):