GPT
S

StableDiffusion-v1-5-Regularization-Images

קוד פתוח

ProGamerGovmit2022-10-31

  • image-text-dataset
  • synthetic-dataset

אוסף תמונות מוכנות מראש לרגולריזציה באימוני דרימבות׳ על סטייבל דיפיוז׳ן 1.5. הוא חוסך שעות של יצירת תמונות רקע עצמאית כדי שהמודל לא ישכח מושגים קיימים.

  • 193הורדות בחודש
  • 165לייקים

מה זה

המאגר כולל עשרות אלפי תמונות סינתטיות ברזולוציה של 512 על 512 פיקסלים, שנשמרו בפורמט PNG. כל התמונות נוצרו במקור מתוך המודל Stable Diffusion v1.5 באמצעות הפיצ׳ר Generate Forever בממשק AUTOMATIC1111, עם ה־VAE המשופר stabilityai/sd-vae-ft-mse. הן מחולקות לקובצי ZIP לפי פרומפטים קבועים, כאשר בכל קובץ שונו רק ערכי ה־seed.

התוכן מאורגן לפי קטגוריות שנועדו לשמש כקלאסים נפוצים: סגנונות אמנות כמו איור או סגנון יצירה, אנשים כולל גברים, נשים ודוגמניות, בעלי חיים כמו חתולים, כלי רכב ונושאים כלליים כמו סייברפאנק. רוב הקבצים הופקו באמצעות דוגם DDIM בחמישים צעדים עם ערך CFG של 7, וחלקם הקטן עם הדוגם DPM++ 2S a Karras ולעיתים עם פרומפט שלילי לטקסט.

מתאים ל

  • אימון דרימבות׳ לאנשים

    שימוש בתמונות איש או אישה כדי לאמן פנים ספציפיות בלי לאבד את היכולת לייצר בני אדם אחרים.

  • הטמעת סגנון אמנותי

    הורדת תמונות מקטגוריית הסגנונות לאימון מודל על ציור מסוים תוך שמירה על המושג הכללי.

  • אימון על בעלי חיים

    שימוש באלפי תמונות חתולים קיימות כקלאס רגולריזציה לאימון חיית מחמד ספציפית.

איפה זה נופל

התמונות נוצרו כולן בסוף שנת 2022 מתוך גרסה 1.5 בלבד, ולכן הן נושאות את כל העיוותים והפגמים האופייניים לגרסה זו, כמו בעיות באנטומיה, ידיים לא מושלמות וארטיפקטים. לא נעשה סינון ידני של התמונות הפגומות, וכולן נשענות על פרומפטים בסיסיים ביותר באנגלית ללא ייצוג מקומי או מגוון רחב. אם אתם מאמנים על בסיס מודלים מודרניים יותר כמו SDXL, שימוש בתמונות רגולריזציה מגרסה 1.5 עלול לפגוע באיכות הפלטים במקום להגן עליהם.

שאלות
נפוצות

האם מותר להשתמש בתמונות האלו באימון מודל למוצר מסחרי?

כן, רישיון MIT מתיר שימוש מסחרי מלא. אתם יכולים לאמן מודלים שיימכרו או ישמשו שירותים בתשלום, בכפוף לשמירת תנאי הרישיון והייחוס של היוצר המקורי.

איך המידע הזה נאסף והאם עבר בקרת איכות?

התמונות נוצרו אוטומטית לחלוטין ברצף מתוך המודל Stable Diffusion v1.5 באמצעות ממשק WebUI. לא נערך שום סינון אנושי, כך שכל הפגמים והעיוותים שהמודל ייצר נכנסו לקובצי הארכיון.

האם הדאטהסט כולל תוכן בעברית או תמונות עם הקשר ישראלי?

לא. כל הפרומפטים ששימשו ליצירה היו מילים בודדות באנגלית כמו car, woman או cyberpunk. אין כאן שום מיקוד תרבותי, שפתי או גאוגרפי מקומי.

מה ההבדל בין הקבצים השונים של אותו הסגנון?

חלק מהקבצים נוצרו עם דוגם DDIM ואחרים עם DPM++ 2S a Karras, וחלקם כללו פרומפט שלילי למניעת טקסט. ההבדל הוא באלגוריתם הדגימה ובמספר התמונות הכולל שנשמר בכל קובץ ZIP.

איך טוענים

אין צורך בקוד מיוחד של Hugging Face כדי לטעון את הנתונים, ואפשר להוריד את קובצי ה־ZIP ישירות מהמאגר ללא דרישת אישור גישה מוקדם. הממשק מציג לעיתים התראה שגויה על קובצי pickle, אך בפועל מדובר בארכיונים פשוטים שכוללים תמונות בלבד. מורידים רק את ה־ZIP שמתאים לקלאס שמאמנים, פותחים אותו ומפנים את סקריפט הדרימבות׳, למשל במימושים של JoePenna או ShivamShrirao, לתיקיית התמונות שחולצו.

from datasets import load_dataset

ds = load_dataset("ProGamerGov/StableDiffusion-v1-5-Regularization-Images")

הרישיון

המאגר מופץ תחת רישיון MIT, שמאפשר שימוש מסחרי, שינוי והפצה כמעט ללא הגבלות, מעבר לדרישת ייחוס בסיסית. המשמעות היא שניתן להשתמש בתמונות בחופשיות לאימון מודלים מסחריים מבלי להידרש לפתוח את הקוד או את משקלי המודל החדש תחת אותו הרישיון.

  • שימוש מסחרי
  • שינוי הקוד
  • הפצה מחדש

הרישיון המלא ב־Hugging Face ↗