Riferimento per ultralytics/data/annotator.py
Nota
Questo file è disponibile all'indirizzo https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/annotator .py. Se riscontri un problema, contribuisci a risolverlo inviando una Pull Request 🛠️. Grazie 🙏!
ultralytics.data.annotator.auto_annotate(data, det_model='yolov8x.pt', sam_model='sam_b.pt', device='', output_dir=None)
Annota automaticamente le immagini utilizzando un modello di rilevamento degli oggetti YOLO e un modello di segmentazione SAM .
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
data |
str
|
Percorso di una cartella contenente le immagini da annotare. |
richiesto |
det_model |
str
|
Modello di rilevamento di YOLO pre-addestrato. Il valore predefinito è 'yolov8x.pt'. |
'yolov8x.pt'
|
sam_model |
str
|
Modello di segmentazione di SAM pre-addestrato. Il valore predefinito è 'sam_b.pt'. |
'sam_b.pt'
|
device |
str
|
Dispositivo su cui eseguire i modelli. Il valore predefinito è una stringa vuota (CPU o GPU, se disponibile). |
''
|
output_dir |
str | None | optional
|
Cartella in cui salvare i risultati annotati. La cartella predefinita è "labels" nella stessa directory di "data". |
None
|