Stable Diffusion - Model Training

ตัวอย่างจะเน้นการฝึกภาพบุคคล

เตรียมภาพเพื่อฝึกโมเดล

สมมุติว่าต้องการฝึกหน้าคน เช่นนาย A ให้เตรียมภาพไว้อย่างน้อย 10-30 รูป ก็เพียงพอโดย

• เน้นภาพคม ๆ คุณภาพสูง noise ไม่เยอะ
• พื้นหลังและสภาพแสงหลาย ๆ แบบ
• หลากหลายมุมไม่ใช่หน้าตรงอย่างเดียว

ภาพยิ่งเยอะยิ่งใช้เวลาฝึกนานแลกกับคุณภาพ

หลังจากนั้นให้ resize ให้มีขนาด 512*512 โดยใช้โปรแกรมที่ถนัด หรือใช้

https://www.birme.net/?target_width=512&target_height=512

โดย resize ให้บุคคลอยู่ตรงกลางภาพหลังจากนั้นจึงจับใส่ folder ให้เรียบร้อยเช่น

D:\AI\Model\ชื่อ folder

TEXTUAL INVERSION Training

https://colab.research.google.com/github/TheLastBen/fast-stable-diffusion/blob/main/fast_stable_diffusion_AUTOMATIC1111.ipynb

• Copy Colab ไปทำเองส่วนตัว
• เชื่อม google drive เราไว้ดึงไฟล์
• Install/Update AUTOMATIC1111 repo (กด run ได้เลย)
  

• Requirements(กด run ได้เลย)

Model Download/Load

• ถ้าไม่มีโมเดลให้ติ๊ก Redownload_the_original_model แล้วกด run ได้เลย
  

Start Stable-Diffusion

• กด run ได้เลยถ้าไม่มีโมเดลมันจะแจ้งเตือนให้ย้อนไปทำขั้นตอนที่แล้ว

แล้วดูไกด์จาก https://www.youtube.com/watch?v=2ityl_dNRNw

ถ้ามี 30 รูป

Batch size 7

Gradient 5

7*5 = 35