هذا هو قالب يمكن استخدامه لتدريب مفاهيم متعددة وصور تنظيم لنماذج LORA (التكيف منخفض الرتبة).
Lora: https://github.com/cloneofsimo/lora
استنادًا إلى البرامج النصية التدريبية بواسطة Kohya-SS SD-Scripts
| دليل | وصف |
|---|---|
image_dir | يحتوي على جميع صور الإدخال وملفات التسمية التوضيحية ، مسبوقة من خلال العدد الإجمالي للعصر لكل صورة |
reg_dir | يحتوي على جميع صور التنظيم لكل مفهوم ، بادئة إجمالي عدد التكرارات لكل صورة Reg |
log | سجلات كل سجل تدريب |
output | الإخراج النهائي الثنائي (Safesensor ، PT ، إلخ) |
config_v1_example.json | التكوين لجلسة التدريب (إعادة تسمية إلى كل ما تريد) |
توازن بين مجموعات البيانات بحيث تشير مجلدات المفاهيم إلى عدد مرات تكرارها أثناء التدريب.
على سبيل المثال ، دعنا نقول أن لديك أربعة مفاهيم وكل منها لديه 14 صورة لكل مفهوم ، باستثناء واحد ، والذي يحتوي على 28 .
لتحقيق التوازن بين خطوات التدريب لكل مفهوم لكل فترة 5200 ، يمكنك تقسيم التكرارات على إجمالي كمية الصور لكل مفهوم:
5200 / 14 = 371 5200 / 28 = 186
في المثال ، سيكون اسم أدلة المفاهيم:
ستضمن الأداة المساعدة أدناه استخدام كل مجلد مفهوم في مجلد مجموعة البيانات بالتساوي أثناء عملية تدريب نموذج التعلم الآلي Dreambooth ، بغض النظر عن عدد الصور في كل مجلد:
https://github.com/bmaltais/kohya_ss/blob/cdf84e2f4b65517c396ffb72474616a57bc109b6/library/dataset_balancing_gui.py