BEN2

מה זה

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

המשקל שלו קל מאוד עם פחות ממאה מיליון פרמטרים, מה שמאפשר לו לרוץ מהר יחסית לפתרונות כבדים יותר. האימון שלו שילב את הדאטה־סט הפתוח DIS5k יחד עם מאגר פרטי של 22 אלף תמונות, במטרה לתת תוצאות טובות גם בחיתוך וידאו וברזולוציות גבוהות של 4K.

מתאים ל

• הסרת רקע בתמונות מוצר

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

• חיתוך שיער בפורטרטים

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

• עיבוד מקדים לווידאו

  משקל קובץ נמוך ומהירות מאפשרים חיתוך פריימים באמצעות FFmpeg ישירות על מכונות פיתוח.

איפה זה נופל

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

איך מריצים

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

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

pip install -U huggingface_hub
hf download PramaLLC/BEN2

הקבצים

16 קבצים במאגר, 1.6 GB סך הכול. הקישורים מובילים ישירות ל־Hugging Face.

• BEN2_Base.pth1.1 GB
• model.safetensors380.6 MB
• BEN2_Base.onnx222.9 MB
• BEN2_demo_pictures/grid_example3.png8.6 MB
• BEN2_demo_pictures/grid_example7.png5.3 MB
• BEN2_demo_pictures/grid_example6.png2.4 MB
• BEN2_demo_pictures/grid_example2.png1.8 MB
• BEN2_demo_pictures/grid_example1.png1.5 MB
• BEN2_demo_pictures/model_comparison.png327.5 KB
• BEN2.py53.5 KB
• README.md5.1 KB
• .gitattributes2.0 KB
• onnx_run.py1.7 KB
• inference.py353 B
• config.json124 B
• requirements.txt102 B

הרישיון

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

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

הרישיון המלא בעמוד המודל ↗