ormbg — הדגמה

מה זה

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

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

מתאים ל

• בידוד אנשים בצילומים

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

• הכנת תמונות לעבודה ברשת

  חיתוך מהיר של נושא התמונה כדי לשלב אותו בפוסטים או באנרים ללא התעסקות בפוטושופ.

• בדיקת איכות של המודל

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

איפה זה נופל

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

איך מריצים

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

pip install -U huggingface_hub
hf download schirrmacher/ormbg

הקבצים

46 קבצים במאגר, 565 MB סך הכול. הקישורים מובילים ישירות ל־Hugging Face.

• models/ormbg.pth176.7 MB
• models/isnet-base-model.pth176.6 MB
• ormbg.onnx176.2 MB
• examples/image/image01_no_background.png21.4 MB
• examples/image/image01.png16.7 MB
• example.png4.8 MB
• examples/example1.png4.8 MB
• dataset/training/im/p_00a5b702.png3.5 MB
• dataset/training/im/p_00a4eda7.png1.7 MB
• examples.jpg1.4 MB
• examples/no-background1.png1.1 MB
• no-background.png1.1 MB
• dataset/validation/im/p_00a7a27c.png1.0 MB
• examples/loss/loss05.png983.2 KB
• examples/loss/loss04.png794.2 KB
• explanation.jpg713.3 KB
• examples/loss/loss03.png644.7 KB
• examples/loss/loss02.png417.3 KB
• examples/loss/orginal.jpg366.0 KB
• examples/loss/loss01.png291.5 KB
• examples/loss/gt.png258.1 KB
• dataset/training/gt/p_00a5b702.png96.9 KB
• dataset/validation/gt/p_00a7a27c.png81.8 KB
• dataset/training/gt/p_00a4eda7.png38.7 KB
• utils/__pycache__/isnet.cpython-312.pyc27.6 KB
• utils/__pycache__/ormbg.cpython-312.pyc20.7 KB
• utils/isnet.py18.5 KB
• ormbg/data_loader_cache.py16.8 KB
• ormbg/train_model.py15.7 KB
• ormbg/models/ormbg.py13.6 KB
• utils/ormbg.py13.3 KB
• examples/.DS_Store6.1 KB
• ormbg/.DS_Store6.1 KB
• utils/.DS_Store6.1 KB
• environment.yaml5.9 KB
• ormbg/inference.py3.3 KB
• README.md2.9 KB
• .gitignore2.8 KB
• utils/inference.py2.6 KB
• .gitattributes2.1 KB

הרישיון

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

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

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