nsfw-image-detection-384
קוד פתוחMarqoapache-2.02024-11-20
זה מסווג תמונות זעיר לזיהוי תוכן למבוגרים ששוקל 22 מגה בייט בלבד. הוא מיועד למי שרוצה סינון מקומי מהיר בלי להחזיק שרת כבד ובלי לשלם לפי קריאת API.
- 6Mפרמטרים
- 22.1 MBמשקל הקבצים
- 84,913הורדות בחודש
- 56לייקים
מה זה
מדובר במודל מבוסס Vision Transformer זעיר, ספציפית כיוונון של vit_tiny_patch16_384, שמקבל תמונות ברזולוציה של 384 על 384 פיקסלים ומחזיר שתי מחלקות בלבד. הוא אומן על מאגר פנימי של 220,000 תמונות שכלל צילומים, איורים, ממים ותוכן שנוצר בבינה מלאכותית, מחציתן מוגדרות כתוכן בטוח ומחציתן לא.
היוצרים שלו מדווחים על דיוק של 98.56% על סט הבדיקה שלהם, מול מתחרים בקוד פתוח כמו הדגמים של AdamCodd או Falconsai. בפועל זה אומר שהצליחו לכווץ מודל פי 18 עד 20 בגודל בלי להפסיד יכולת זיהוי בסיסית, כל עוד ההגדרה שלכם לתוכן בעייתי חופפת לדאטה שבו השתמשו.
מתאים ל
- סינון ראשוני בצד השרת
בדיקת תמונות שמשתמשים מעלים ישירות על המעבד, בלי לעכב את הבקשה ובלי עלויות GPU יקרות.
- ניקוי מאגרי מידע לאימון
מעבר אוטומטי על מיליוני תמונות ממקורות פתוחים כדי להעיף תוכן בוטה לפני שמאמנים מודלים אחרים.
- בדיקה במכשירי קצה
בזכות משקל של 22 מגה בייט אפשר לדחוף אותו לאפליקציות שולחניות או מערכות פנימיות בלי לפגוע בביצועים.
איפה זה נופל
ההגדרה של תוכן לא הולם היא עניין תלוי הקשר ותרבות, והמודל נותן רק תשובה בינארית של כן או לא. היוצרים עצמם מזהירים שהוא עלול לטעות, ושחובה לשחק עם סף ההסתברות כדי לאזן בין תמונות תמימות שנחסמות לבין תוכן בוטה שחומק. בנוסף, הדאטהסט המלא שבו אומן נשאר קנייני וסגור, כך שקשה לדעת מראש אילו מקרי קצה ספציפיים יכשילו אותו בלי לבדוק על התמונות שלכם.
שאלות
נפוצות
האם המודל מפרט איזה סוג תוכן בעייתי יש בתמונה?
לא. המודל מחזיר רק סיווג בינארי בין תוכן בטוח לתוכן לא בטוח. אם אתם צריכים לדעת האם מדובר באלימות, עירום או סתם מם גבולי, תצטרכו מודל אחר או כלי שבודק כמה קטגוריות במקביל.
חייבים GPU כדי להשתמש בו בסביבת ייצור?
ממש לא. הוא שוקל 22.1 מגה בייט וכולל כשישה מיליון פרמטרים בלבד, שזה גודל שמעבדי CPU רגילים מריצים באלפיות שניה. כרטיס מסך יידרש רק אם אתם צריכים לעבד אלפי תמונות בשנייה במקביל.
איך מריצים
טוענים אותו ישירות דרך ספריית timm בפייתון בשלוש שורות קוד, ומעבירים אליו תמונה שעברה עיבוד מקדים תואם. עם פחות משישה מיליון פרמטרים ומשקל קובץ מגוחך, אין שום צורך בכרטיס מסך ייעודי. הוא ירוץ על כל מעבד סטנדרטי בשרת ענן זול או אפילו במכשיר קצה, ומחזיר הסתברות שאפשר להמיר לסיווג.
זה אומר שאין סיבה לקרוא ל־API חיצוני אלא אם אתם לא רוצים לתחזק קוד בכלל, או שהשירות המרוחק נותן לכם חלוקה מפורטת לקטגוריות שונות שאין כאן.
pip install -U huggingface_hub hf download Marqo/nsfw-image-detection-384
הקבצים
7 קבצים במאגר, 22.1 MB סך הכול. הקישורים מובילים ישירות ל־Hugging Face.
הרישיון
הרישיון הוא Apache 2.0. מותר להשתמש בו לצרכים מסחריים, לשנות אותו ולשלב אותו במוצרים פנימיים או חיצוניים בלי לשלם תמלוגים, בתנאי ששומרים על הודעת זכויות היוצרים והרישיון המקורי.
- שימוש מסחרי
- שינוי הקוד
- הפצה מחדש
- שימוש בפטנטים