مرجع ل ultralytics/models/sam/model.py
ملاحظه
هذا الملف متاح في https://github.com/ultralytics/ultralytics/ نقطة / الرئيسية /ultralytics/نماذج/sam/model.py. إذا اكتشفت مشكلة ، فيرجى المساعدة في إصلاحها من خلال المساهمة في طلب 🛠️ سحب. شكرا لك 🙏!
ultralytics.models.sam.model.SAM
قواعد: Model
SAM (نموذج جزء أي شيء) فئة الواجهة.
SAM تم تصميمه لتجزئة الصور في الوقت الفعلي بشكل سريع. يمكن استخدامه مع مجموعة متنوعة من المطالبات مثل المربعات المحيطة أو النقاط أو التسميات. يتمتع النموذج بقدرات لأداء طلقة صفرية ويتم تدريبه على SA-1B مجموعة البيانات.
شفرة المصدر في ultralytics/models/sam/model.py
task_map
property
يوفر تعيينا من مهمة "الجزء" إلى "المتنبئ" المقابل لها.
ارجاع:
نوع | وصف |
---|---|
dict
|
قاموس يعين مهمة "الجزء" إلى "المتنبئ" المقابل لها. |
__call__(source=None, stream=False, bboxes=None, points=None, labels=None, **kwargs)
الاسم المستعار لطريقة "التنبؤ".
البارامترات:
اسم | نوع | وصف | افتراضي |
---|---|---|---|
source |
str
|
المسار إلى ملف الصورة أو الفيديو أو PIL. كائن صورة، أو كائن numpy.ndarray . |
None
|
stream |
bool
|
إذا كان صحيحا ، يتيح البث في الوقت الفعلي. الإعدادات الافتراضية إلى خطأ. |
False
|
bboxes |
list
|
قائمة إحداثيات المربع المحيط للتجزئة المطلوبة. الإعدادات الافتراضية إلى لا شيء. |
None
|
points |
list
|
قائمة النقاط للتجزئة المطلوبة. الإعدادات الافتراضية إلى لا شيء. |
None
|
labels |
list
|
قائمة التسميات للتجزئة المطلوبة. الإعدادات الافتراضية إلى لا شيء. |
None
|
ارجاع:
نوع | وصف |
---|---|
list
|
تنبؤات النموذج. |
شفرة المصدر في ultralytics/models/sam/model.py
__init__(model='sam_b.pt')
تهيئة SAM نموذج مع ملف نموذج مدرب مسبقا.
البارامترات:
اسم | نوع | وصف | افتراضي |
---|---|---|---|
model |
str
|
الطريق إلى المدربين مسبقا SAM ملف النموذج. يجب أن يكون للملف ملحق .pt أو .pth. |
'sam_b.pt'
|
يثير:
نوع | وصف |
---|---|
NotImplementedError
|
إذا لم يكن ملحق ملف النموذج .pt أو .pth. |
شفرة المصدر في ultralytics/models/sam/model.py
info(detailed=False, verbose=True)
يسجل معلومات حول SAM نموذج.
البارامترات:
اسم | نوع | وصف | افتراضي |
---|---|---|---|
detailed |
bool
|
إذا كان True ، يعرض معلومات تفصيلية حول النموذج. الإعدادات الافتراضية إلى خطأ. |
False
|
verbose |
bool
|
إذا كان صحيحا، يعرض المعلومات على وحدة التحكم. الإعدادات الافتراضية إلى صواب. |
True
|
ارجاع:
نوع | وصف |
---|---|
tuple
|
مجموعة تحتوي على معلومات النموذج. |
شفرة المصدر في ultralytics/models/sam/model.py
predict(source, stream=False, bboxes=None, points=None, labels=None, **kwargs)
يقوم بالتنبؤ بالتجزئة على الصورة أو مصدر الفيديو المحدد.
البارامترات:
اسم | نوع | وصف | افتراضي |
---|---|---|---|
source |
str
|
المسار إلى ملف الصورة أو الفيديو أو PIL. كائن صورة، أو كائن numpy.ndarray . |
مطلوب |
stream |
bool
|
إذا كان صحيحا ، يتيح البث في الوقت الفعلي. الإعدادات الافتراضية إلى خطأ. |
False
|
bboxes |
list
|
قائمة إحداثيات المربع المحيط للتجزئة المطلوبة. الإعدادات الافتراضية إلى لا شيء. |
None
|
points |
list
|
قائمة النقاط للتجزئة المطلوبة. الإعدادات الافتراضية إلى لا شيء. |
None
|
labels |
list
|
قائمة التسميات للتجزئة المطلوبة. الإعدادات الافتراضية إلى لا شيء. |
None
|
ارجاع:
نوع | وصف |
---|---|
list
|
تنبؤات النموذج. |