Source code for vlkit.io.bytes

import numpy as np
from io import BytesIO
import PIL
from PIL import Image


[docs]def bytes2array(b: bytes) -> np.ndarray:
    return np.load(BytesIO(b), allow_pickle=True)


[docs]def bytes2image(b: bytes) -> PIL.Image.Image:
    return Image.open(BytesIO(b))


[docs]def bytes2image2array(b: bytes) -> np.ndarray:
    return np.array(bytes2image(b))


[docs]def array2bytes(b: bytes) -> bytes:
    """
    save ndarray shape information into bytes
    refer to https://stackoverflow.com/a/61838233/4095689
    """
    np_bytes = BytesIO()
    np.save(np_bytes, b, allow_pickle=True)
    return np_bytes.getvalue()


[docs]def image2bytes(image: PIL.Image.Image) -> bytes:
    assert isinstance(image, Image.Image)
    b = BytesIO()
    image.save(b)
    return b.getvalue()