Referans için ultralytics/models/sam/model.py
Not
Bu dosya https://github.com/ultralytics/ultralytics/blob/main/ ultralytics/models/ sam/model .py adresinde mevcuttur. Bir sorun tespit ederseniz lütfen bir Çekme İsteği 🛠️ ile katkıda bulunarak düzeltilmesine yardımcı olun. Teşekkürler 🙏!
ultralytics.models.sam.model.SAM
Üsler: Model
SAM (Segment Anything Model) arayüz sınıfı.
SAM istenebilir gerçek zamanlı görüntü segmentasyonu için tasarlanmıştır. Aşağıdakiler gibi çeşitli istemlerle kullanılabilir sınırlayıcı kutular, noktalar veya etiketler. Model sıfır atış performansı için yeteneklere sahiptir ve SA-1B üzerinde eğitilmiştir veri kümesi.
Kaynak kodu ultralytics/models/sam/model.py
task_map
property
'Segment' görevinden karşılık gelen 'Tahmin Edici'ye bir eşleme sağlar.
İade:
Tip | Açıklama |
---|---|
dict
|
'Segment' görevini karşılık gelen 'Tahmin Edici' ile eşleyen bir sözlük. |
__call__(source=None, stream=False, bboxes=None, points=None, labels=None, **kwargs)
'Tahmin' yöntemi için takma ad.
Parametreler:
İsim | Tip | Açıklama | Varsayılan |
---|---|---|---|
source |
str
|
Görüntü veya video dosyasının yolu veya bir PIL.Image nesnesi ya da bir numpy.ndarray nesnesi. |
None
|
stream |
bool
|
True ise, gerçek zamanlı akışı etkinleştirir. Varsayılan değer False'dir. |
False
|
bboxes |
list
|
İstenen segmentasyon için sınırlayıcı kutu koordinatlarının listesi. Varsayılan değer Yok'tur. |
None
|
points |
list
|
İstenen segmentasyon için noktaların listesi. Varsayılan değer Yok'tur. |
None
|
labels |
list
|
İstenen segmentasyon için etiketlerin listesi. Varsayılan değer Yok'tur. |
None
|
İade:
Tip | Açıklama |
---|---|
list
|
Model tahminleri. |
Kaynak kodu ultralytics/models/sam/model.py
__init__(model='sam_b.pt')
SAM modelini önceden eğitilmiş bir model dosyası ile başlatır.
Parametreler:
İsim | Tip | Açıklama | Varsayılan |
---|---|---|---|
model |
str
|
Önceden eğitilmiş SAM model dosyasının yolu. Dosya .pt veya .pth uzantısına sahip olmalıdır. |
'sam_b.pt'
|
Zamlar:
Tip | Açıklama |
---|---|
NotImplementedError
|
Model dosya uzantısı .pt veya .pth değilse. |
Kaynak kodu ultralytics/models/sam/model.py
info(detailed=False, verbose=True)
SAM modeli hakkındaki bilgileri günlüğe kaydeder.
Parametreler:
İsim | Tip | Açıklama | Varsayılan |
---|---|---|---|
detailed |
bool
|
True ise, model hakkında ayrıntılı bilgi görüntüler. Varsayılan değer False'dir. |
False
|
verbose |
bool
|
True ise, bilgileri konsolda görüntüler. Varsayılan değer True'dur. |
True
|
İade:
Tip | Açıklama |
---|---|
tuple
|
Modelin bilgilerini içeren bir tuple. |
Kaynak kodu ultralytics/models/sam/model.py
predict(source, stream=False, bboxes=None, points=None, labels=None, **kwargs)
Verilen görüntü veya video kaynağı üzerinde segmentasyon tahmini gerçekleştirir.
Parametreler:
İsim | Tip | Açıklama | Varsayılan |
---|---|---|---|
source |
str
|
Görüntü veya video dosyasının yolu veya bir PIL.Image nesnesi ya da bir numpy.ndarray nesnesi. |
gerekli |
stream |
bool
|
True ise, gerçek zamanlı akışı etkinleştirir. Varsayılan değer False'dir. |
False
|
bboxes |
list
|
İstenen segmentasyon için sınırlayıcı kutu koordinatlarının listesi. Varsayılan değer Yok'tur. |
None
|
points |
list
|
İstenen segmentasyon için noktaların listesi. Varsayılan değer Yok'tur. |
None
|
labels |
list
|
İstenen segmentasyon için etiketlerin listesi. Varsayılan değer Yok'tur. |
None
|
İade:
Tip | Açıklama |
---|---|
list
|
Model tahminleri. |