refactor: share backend image serialization helpers

This commit is contained in:
Dariusz L
2026-08-13 05:39:24 +02:00
parent f60b7c0d8d
commit 212d6b7728
5 changed files with 170 additions and 55 deletions
+2 -7
View File
@@ -1,22 +1,17 @@
"""Image and tensor conversion helpers shared by LayerForge endpoints."""
import base64
import io
import numpy as np
from PIL import Image
from .image_serialization import pil_to_data_url
from torchvision import transforms
from .image_serialization import data_url_to_pil, pil_to_data_url
from .node import log
def convert_base64_to_tensor(base64_str):
"""Convert a data-URL image into a BCHW tensor and optional alpha tensor."""
try:
img_data = base64.b64decode(base64_str.split(",")[1])
img = Image.open(io.BytesIO(img_data))
img = data_url_to_pil(base64_str)
has_alpha = img.mode == "RGBA"
alpha = None