ちょっと前に世間を騒がせた画像生成AI, Stable Diffusion ですが、Kerasの公式ページで実装が公開されているのを知りました。これだと高価なPCを持っていなくても、google colaboratory で簡単に試すことができます。
High-performance image generation using Stable Diffusion in KerasCV
ここから[View in Colab]で、colabに入り、各セルの実行ボタンを押していけば簡単に、画像を生成するところまで行けます。
当然ですが、コード中のプロンプトは自由に変えられます。
images = model.text_to_image(
"Realistic and colorful photos, "
"Adventurers are holding the latest weapons hidden in jungle ruins, ",
batch_size=3,
)
plot_images(images)
Stable Diffusion のコードもこちらで見れます。
stable diffusion のlatent spaceを探索してみるという記事もあり、至れり尽くせりです。
コメントを残す