Stable Diffusion で prompt 情報を画像に保存しておく | stablediffusion | tech - 氾濫原

PNG にはメタデータを保存するチャンクがある。Python からも簡単に読み書き可能なので、ここに prompt を保存しておくと、あとから参照したいときに便利になる img が PIL の Image だとして、save を呼びだしているとこに pnginfo というのを足してあげる。 from PIL.PngImagePlugin import PngInfo # ... metadata = PngInfo() metadata.add_text("prompt", "foo bar baz") img.save("./outputs/0.png", pnginfo=metadata) 読み出し #!/usr/bin/env python import sys from PIL import Image from PIL.PngImagePlugin import PngInfo

[june29]

ほー、PNG にはメタデータを埋め込めるのか。