cv_backbones
קוד פתוחmonetjoemit2023-04-27
אפשר גם להריץ אותו בדפדפן בלי להתקין דבר.
זה לא מאגר תמונות אלא אינדקס קישורים ומטא דאטה של מודלי ראייה ממוחשבת מפייטורץ'. הוא חוסך חיפוש ידני של משקולות ומאפייני ארכיטקטורה לאימון העברה.
- 125הורדות בחודש
- 27לייקים
מה זה
המאגר מרכז רשימה של מודלי שלד לראייה ממוחשבת שאומנו מראש על ImageNet1K ונלקחו ישירות מהאתר הרשמי של פייטורץ'. הרשומות כוללות רשתות קונבולוציה קלאסיות כמו רזנט או VGG לצד מודלי טרנספורמר כמו Swin ו־ViT.
כל שורה מייצגת מודל בודד עם ארבעה שדות פשוטים: שם הגרסה, סוג הארכיטקטורה, ממדי תמונת הקלט, וקישור ישיר לקובץ המשקולות בפורמט pth. אין פה נתונים ויזואליים גולמיים אלא מפת הפניות למשקולות חיצוניות.
התוכן מחולק לשני חלקים לפי גרסאות האימון של פייטורץ'. חלק הטריין מכיל מודלים שאומנו על הגרסה הראשונה של אימג'נט, וחלק הטסט מוקדש למודלים שרצים על גרסת האימון השנייה והעדכנית יותר.
מתאים ל
- סריקה והורדה אוטומטית
משיכת קישורים ישירים לקובצי pth של משקולות מודלים לפי גודל קלט וסוג ארכיטקטורה.
- השוואת מורכבות ארכיטקטורות
מיפוי גודלי מודלים ומספר פרמטרים לצורך בחירת שלד מתאים ליישומי קצה מוגבלי משאבים.
- הקמת תשתיות אימון העברה
שילוב כתובות המשקולות בצנרת פיתוח אוטומטית שבוחרת בין גרסאות שונות של אימג'נט.
איפה זה נופל
הנתונים במאגר תלויים לחלוטין בשרתים החיצוניים של פייטורץ', ואם כתובת של קובץ משקולות משתנה או נמחקת, השורה הופכת לחסרת תועלת. המאגר עודכן באפריל 2023 ומכיל רק מה שהיה זמין רשמית בפייטורץ' באותה נקודה, בלי ארכיטקטורות חדשות יותר.
השפה המוגדרת היא אנגלית בלבד ומדובר במטא דאטה טכני נטו. אם אתם מחפשים תמונות לאימון או קוד מוכן להרצה, לא תמצאו אותם כאן, אלא רק מפתח להורדת משקולות שאומנו על תמונות אימג'נט.
שאלות
נפוצות
האם יש במאגר תמונות שניתן לאמן עליהן ישירות?
לא. המאגר מכיל רק קובצי טקסט מסוג jsonl עם שמות מודלים, גודל קלט נדרש וקישורים להורדת המשקולות. הוא לא כולל פיקסלים או מערכי נתונים ויזואליים.
האם מותר להשתמש במאגר לפיתוח מוצר מסחרי?
רשימת המידע עצמה ברישיון MIT ופתוחה לכל שימוש. עם זאת, המשקולות שהקישורים מובילים אליהן שייכות לפייטורץ' ואומנו על אימג'נט, ולכן נדרש לבדוק את תנאי הרישיון של המשקולות עצמן מול פייטורץ'.
כמה מקום המאגר תופס בדיסק?
המאגר כולל פחות מאלף רשומות בחמישה קבצים בלבד, ושוקל פחות ממגהבייט בודד. מה שתופס נפח אמיתי הוא המשקולות שאותן תורידו בנפרד מתוך הקישורים שבפנים.
מה ההבדל בין החלוקה לטריין וטסט כאן?
החלוקה אינה משמשת לאימון ובדיקה רגילים של למידת מכונה. חלק הטריין מרכז מודלים שאומנו במקור על הגרסה הראשונה של אימג'נט, בעוד חלק הטסט מרכז את המודלים שאומנו על הגרסה השנייה.
איך טוענים
טוענים את הרשימה באמצעות load_dataset של ספריית datasets ומציינים את הפיצול הרצוי, train לגרסה הראשונה או test לשנייה. המאגר מכיל פחות מאלף רשומות בקובצי jsonl, כך שההורדה שלו מיידית ואינה דורשת אישורי גישה או מקום בדיסק מעבר לכמה קילובייטים.
קחו בחשבון שהמאגר נותן קישורים בלבד. כדי להשתמש במשקולות עצמן תצטרכו להוריד את קובצי ה־pth ישירות מהשרתים של פייטורץ', ושם כבר מדובר בנפחים שמגיעים למאות מגהבייטים לכל מודל.
from datasets import load_dataset
ds = load_dataset("monetjoe/cv_backbones")הרישיון
המאגר מופץ תחת רישיון MIT המאפשר שימוש מסחרי, שינוי והפצה ללא הגבלה, בתנאי ששומרים על הודעת זכויות היוצרים המקורית. הרישיון מכסה את קובצי המידע שבמאגר עצמו. אם אתם מורידים את המשקולות מהקישורים ומאמנים עליהן מודל, עליכם לבדוק בנפרד את תנאי השימוש ברישיון של פייטורץ' ושל אימג'נט לגבי שימוש מסחרי במשקולות.
- שימוש מסחרי
- שינוי הקוד
- הפצה מחדש