24 lines
655 B
Python
24 lines
655 B
Python
import re
|
|
import base64
|
|
import numpy as np
|
|
from PIL import Image
|
|
from io import BytesIO
|
|
|
|
def base64_to_pil(img_base64):
|
|
"""
|
|
Convert base64 image data to PIL image
|
|
"""
|
|
image_data = re.sub('^data:image/.+;base64,', '', img_base64)
|
|
pil_image = Image.open(BytesIO(base64.b64decode(image_data)))
|
|
return pil_image
|
|
|
|
def np_to_base64(img_np):
|
|
"""
|
|
Convert numpy image (RGB) to base64 string
|
|
"""
|
|
img = Image.fromarray(img_np.astype('uint8'), 'RGB')
|
|
buffered = BytesIO()
|
|
img.save(buffered, format="PNG")
|
|
img_str = base64.b64encode(buffered.getvalue()).decode("ascii")
|
|
return f"data:image/png;base64,{img_str}"
|