mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-10 06:10:50 +08:00
23 lines
526 B
Python
23 lines
526 B
Python
import os.path
|
|
from contextlib import contextmanager
|
|
from typing import Iterator
|
|
|
|
import cv2
|
|
from PIL import Image
|
|
|
|
from . import node_helpers
|
|
|
|
|
|
def _open_exr(exr_path) -> Image.Image:
|
|
return Image.fromarray(cv2.imread(exr_path, cv2.IMREAD_COLOR))
|
|
|
|
|
|
@contextmanager
|
|
def open_image(file_path: str) -> Iterator[Image.Image]:
|
|
_, ext = os.path.splitext(file_path)
|
|
if ext == ".exr":
|
|
yield _open_exr(file_path)
|
|
else:
|
|
with node_helpers.pillow(Image.open, file_path) as image:
|
|
yield image
|