xiaden
315666c7b6
- Split flat node modules into categorized subpackages (tagging, detect, edge, pose, segment, restore, compare, judge, transform, utility) - Extract shared internals into _shared/ (bases, tensor, formatting) - Centralize node registration via register_nodes() helper - Clean up: remove dead is_scored return, unnecessary hasattr guard, inline imports
30 lines
906 B
Python
30 lines
906 B
Python
"""
|
|
ComfyUI IMAGE tensor <-> PIL.Image conversion utilities.
|
|
"""
|
|
|
|
import numpy as np
|
|
from PIL import Image
|
|
|
|
|
|
def comfy_to_pil(image_tensor) -> Image.Image:
|
|
"""Convert a ComfyUI IMAGE tensor to a PIL Image. Only the first image in the batch is processed."""
|
|
img = image_tensor.numpy()[0].copy()
|
|
img = (img * 255.0).clip(0, 255).astype(np.uint8)
|
|
return Image.fromarray(img, mode="RGB")
|
|
|
|
|
|
def pil_to_comfy(pil_image: Image.Image) -> np.ndarray:
|
|
"""Convert a PIL Image to a ComfyUI IMAGE tensor."""
|
|
if pil_image.mode not in ("RGB", "RGBA", "L", "1"):
|
|
pil_image = pil_image.convert("RGB")
|
|
|
|
if pil_image.mode in ("L", "1"):
|
|
img = np.array(pil_image, dtype=np.float32) / 255.0
|
|
img = img[:, :, np.newaxis]
|
|
img = img[np.newaxis, ...]
|
|
else:
|
|
img = np.array(pil_image, dtype=np.float32) / 255.0
|
|
img = img[np.newaxis, ...]
|
|
|
|
return img
|