Skip to main content

Stable Diffusion - Model Training

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

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

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

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

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

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

     

     

     



    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

    image.png