Skip to content

pixano.datasets.utils.image

image_to_thumbnail(image)

Generate image thumbnail.

Parameters:

Name Type Description Default
image bytes | Image

Image as binary or as a Pillow Image.

required

Returns:

Type Description
bytes

Image thumbnail as binary.

Source code in pixano/datasets/utils/image.py
def image_to_thumbnail(image: bytes | Image.Image) -> bytes:
    """Generate image thumbnail.

    Args:
        image: Image as binary or as a Pillow Image.

    Returns:
        Image thumbnail as binary.
    """
    if isinstance(image, bytes):
        image = Image.open(BytesIO(image))

    image.thumbnail((128, 128))
    return image_to_binary(image)