Referenz für ultralytics/data/annotator.py
Hinweis
Diese Datei ist verfügbar unter https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/annotator .py. Wenn du ein Problem entdeckst, hilf bitte, es zu beheben, indem du einen Pull Request 🛠️ einreichst. Vielen Dank 🙏!
ultralytics.data.annotator.auto_annotate(data, det_model='yolov8x.pt', sam_model='sam_b.pt', device='', output_dir=None)
Beschriftet Bilder automatisch mit einem YOLO Objekterkennungsmodell und einem SAM Segmentierungsmodell.
Parameter:
Name | Typ | Beschreibung | Standard |
---|---|---|---|
data |
str
|
Pfad zu einem Ordner mit Bildern, die mit Anmerkungen versehen werden sollen. |
erforderlich |
det_model |
str
|
Vorgeprüftes YOLO Erkennungsmodell. Die Voreinstellung ist 'yolov8x.pt'. |
'yolov8x.pt'
|
sam_model |
str
|
Vortrainiertes SAM Segmentierungsmodell. Der Standardwert ist 'sam_b.pt'. |
'sam_b.pt'
|
device |
str
|
Gerät, auf dem die Modelle ausgeführt werden. Der Standardwert ist ein leerer String (CPU oder GPU, falls verfügbar). |
''
|
output_dir |
str | None | optional
|
Verzeichnis zum Speichern der kommentierten Ergebnisse. Standardmäßig wird der Ordner "labels" im selben Verzeichnis wie "data" gespeichert. |
None
|