1
1

More than 1 year has passed since last update.

Diffusion Model のおもしろい技術:DreamBoothについて

Last updated at Posted at 2023-09-13

DreamBoothとは、対象を特定して画像を生成させる技術です。
まずは下の図のように、対象(犬)の写真を複数枚(色んな角度からの)を用意してモデルに学習させて、
その後はプロンプトによって対象が実際に行ったことがない画像も生成できます。
「getting a haircut」と指定すれば、対象(犬)が髪を切ってるような画像を作れてくれます。
1.PNG

他にも色々機能があって、例えば下のように対象(犬)と同じ顔になっているパンダ、ライオンの画像など、
こういう複雑なプロンプトを指定しても問題なく生成できます。2.PNG

ネットで探した猫の写真と"猫がバケツにいる"というプロンプトを使って試してみましたが、1枚だけでもかなりきれいな画像ができました。
3.jpg
4.jpg

興味ある方は下のコードとホームページなどを参考してください。
コード:
https://colab.research.google.com/github/ShivamShrirao/diffusers/blob/main/examples/dreambooth/DreamBooth_Stable_Diffusion.ipynb#scrollTo=wnTMyW41cC1E
ホームページ:
https://dreambooth.github.io/

1
1
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
1
1