giers10/skymap-gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

auto-git:

Browse Source 
[change] generate_equirect.py
This commit is contained in:
Victor Giers
2026-05-07 11:03:33 +02:00
parent fd98d255b0
commit 6fdbd9d94a
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
generate_equirect.py
View File

@@ -16,6 +16,7 @@ import shutil
import tempfile import tempfile
import torch import torch
from PIL import Image, ImageDraw from PIL import Image, ImageDraw
from PIL.PngImagePlugin import PngInfo
import numpy as np import numpy as np
from diffusers import ( from diffusers import (
@@ -62,6 +63,18 @@ def next_filename(output_dir: str, base: str, width: int, height: int) -> str:
i += 1 i += 1
def save_png_with_prompt(img: Image.Image, out_path: str, prompt: str) -> None:
pnginfo = PngInfo()
pnginfo.add_text("prompt", prompt)
save_kwargs = {"pnginfo": pnginfo}
for key in ("icc_profile", "exif", "dpi"):
if key in img.info:
save_kwargs[key] = img.info[key]
img.save(out_path, **save_kwargs)
def shift_image(img: Image.Image, shift: int) -> Image.Image: def shift_image(img: Image.Image, shift: int) -> Image.Image:
w, h = img.size w, h = img.size
out = Image.new("RGB", (w, h)) out = Image.new("RGB", (w, h))