Stable Diffusion : Dreambooth로 무료 셀프 AI Avatar 만들어보기

이 사진들은 모두 AI가 만든 사진입니닷

0. 학습시킬 내 사진을 준비합니다

사진 데이터의 개수는 6개 ~ 16개 까지 해보았는데 다 퀄이 괜찮았습니다

얼굴 이미지를 학습에 사용하기 위해서
512*512px로 사이즈를 수정하고,
파일명은 고유한 변수 + 넘버링 으로 했습니다

e.g. "johamcarat (1)" (이세상에 존재하지 않는 단어인것이 좋음)

사이즈 조정 및 네이밍 변경은
아래 Birme 사이트에서 간편하게 가능합니다

BIRME - Bulk Image Resizing Made Easy 2.0 (Online & Free)

 

아래 colab 노트북을 열어줍니다

https://colab.research.google.com/github/sagiodev/stablediffusion_webui/blob/master/DreamBooth_Stable_Diffusion_SDA.ipynb

DreamBooth_Stable_Diffusion_SDA.ipynb

 

1. 모델 학습시키기 

첫 실행 블록입니다. 
변경할 것은 두가지 입니다

zwx를 본인의 고유한 변수 (e.g. johamcarat) 으로 변경해주세요
toy를 person으로 변경해주세요

그대로 실행하면 중간에 준비한 이미지 파일들을 업로드 하고,
약 20분 정도 소요되며
이 동안 모델에 제 얼굴이 학습됩니다..

다 학습이 되고 나온 샘플 이미지들. 아직까지는 나를 그렇게 닮은지 모르겠지만..

 

2. 모델에 prompt 넣어서 text-to-image 생성해보기

고유한 변수 "johamcarat" 을 사용하여 모델에 prompt를 넣어보겠습니다
(변수를 사용하지 않아도 이미지는 생성됩니다만,, 
제 자신의 아바타를 보기 위해서는 꼭 고유한 변수를 사용해주어야 합니다)

e.g. portrait of johamcarat, 8k, high quality, k-pop idol

 

3. 더 나은 결과물을 얻기 위한 prompt 찾아 헤메기

prompt 에 따라 결과물이 천차만별입니다
negative prompt 도 잘 사용하시면 원하시는 결과물에 더 가까워 지실거에요

몇가지 맘에 들었던 prompt 공유합니다

- pencil sketch, portrait of johamcarat woman, high quality, Gray-scale
- portrait of johamcarat woman, high quality, watercolor, detailed
- portrait of johamcarat person, 8k, high quality
- pencil sketch of johamcarat woman, joham, joham woman, inpired by greg rutkowski, digital art by artgemGuidance

 

참고

https://stable-diffusion-art.com/dreambooth/