Function numpy_to_bytes

vision_agent/utils/image_utils.py:122–128

(image: np.ndarray)

def numpy_to_bytes(image: np.ndarray) -> bytes:
    pil_image = Image.fromarray(image).convert("RGB")
    image_buffer = io.BytesIO()
    pil_image.save(image_buffer, format="PNG")
    buffer_bytes = image_buffer.getvalue()
    image_buffer.close()
    return buffer_bytes


def get_image_size(data: Union[str, Path, np.ndarray, ImageType]) -> Tuple[int, ...]: