Referentie voor ultralytics/data/annotator.py
Opmerking
Dit bestand is beschikbaar op https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/annotator .py. Als je een probleem ziet, help het dan oplossen door een Pull Request 🛠️ bij te dragen. Bedankt 🙏!
ultralytics.data.annotator.auto_annotate(data, det_model='yolov8x.pt', sam_model='sam_b.pt', device='', output_dir=None)
Annoteert afbeeldingen automatisch met behulp van een YOLO objectdetectiemodel en een SAM segmentatiemodel.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
data |
str
|
Pad naar een map met afbeeldingen die geannoteerd moeten worden. |
vereist |
det_model |
str
|
Voorgetraind YOLO detectiemodel. Standaard ingesteld op 'yolov8x.pt'. |
'yolov8x.pt'
|
sam_model |
str
|
Voorgetraind SAM segmentatiemodel. Standaard ingesteld op 'sam_b.pt'. |
'sam_b.pt'
|
device |
str
|
Apparaat om de modellen op uit te voeren. Standaard een lege string (CPU of GPU, indien beschikbaar). |
''
|
output_dir |
str | None | optional
|
Directory om de geannoteerde resultaten op te slaan. Standaard een map 'labels' in dezelfde map als 'gegevens'. |
None
|