Referans için ultralytics/data/annotator.py
Not
Bu dosya https://github.com/ultralytics/ultralytics/blob/main/ ultralytics/data/annotator .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.data.annotator.auto_annotate(data, det_model='yolov8x.pt', sam_model='sam_b.pt', device='', output_dir=None)
Bir YOLO nesne algılama modeli ve bir SAM segmentasyon modeli kullanarak görüntülere otomatik olarak açıklama ekler.
Parametreler:
İsim | Tip | Açıklama | Varsayılan |
---|---|---|---|
data |
str
|
Açıklama eklenecek görüntüleri içeren bir klasörün yolu. |
gerekli |
det_model |
str
|
Önceden eğitilmiş YOLO algılama modeli. Varsayılan değer 'yolov8x.pt'. |
'yolov8x.pt'
|
sam_model |
str
|
Önceden eğitilmiş SAM segmentasyon modeli. Varsayılan değer 'sam_b.pt'. |
'sam_b.pt'
|
device |
str
|
Modellerin çalıştırılacağı cihaz. Varsayılan olarak boş bir dizedir (varsa CPU veya GPU). |
''
|
output_dir |
str | None | optional
|
Açıklamalı sonuçların kaydedileceği dizin. Varsayılan olarak 'data' ile aynı dizinde bir 'labels' klasörüdür. |
None
|