Stable Diffusion で prompt 情報を画像に保存しておく
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
img = Image.open(sys.argv[1])
print(img.text['prompt'])
関連エントリー
- daemontoolsのsetuidgidが補助グループ (supplementary groups) 権限をつけてくれない問題 (python) いままでハマったことがなかったのだけど、ついにハマってしまった。 補助グループ権限もつけてくれるsetuidgidのようなもの - (ひ)メ...
- Raspberry Pi から HDMI-CEC コントロール サイネージ HDMI CEC 連携 カレンダーを表示するおうちサイネージ というのを作っていたがモニタの置き場がなくなってしまったので、テレ...
- NanoPi NEO2 でやること、NanoHat OLED のメモ書き nanopi-neo2_friendlycore-xenial_4.14.52_20180628.img.zip の場合 /etc/rc.l...
- RNN/LSTM/GRU の入力と stateful 化 (keras) 入力データ (batch_size, timesteps, features) が入力になる。batch_size はこのバッチ(学習・予測...
- builderscon の電子名札に家庭のメトリクス表示 やってること prometheus の API を叩く 表示する だけ。 表示部分はせっかく python の骨格があるので基本的にはそのま...