Référence pour ultralytics/data/annotator.py
Note
Ce fichier est disponible à l'adresse https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/annotator .py. Si tu repères un problème, aide à le corriger en contribuant à une Pull Request 🛠️. Merci 🙏 !
ultralytics.data.annotator.auto_annotate(data, det_model='yolov8x.pt', sam_model='sam_b.pt', device='', output_dir=None)
Annote automatiquement les images à l'aide d'un modèle de détection d'objets YOLO et d'un modèle de segmentation SAM .
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
data |
str
|
Chemin d'accès à un dossier contenant des images à annoter. |
requis |
det_model |
str
|
Modèle de détection de YOLO pré-entraîné. La valeur par défaut est 'yolov8x.pt'. |
'yolov8x.pt'
|
sam_model |
str
|
Modèle de segmentation pré-entraîné SAM . La valeur par défaut est 'sam_b.pt'. |
'sam_b.pt'
|
device |
str
|
Périphérique sur lequel les modèles sont exécutés. La valeur par défaut est une chaîne vide (CPU ou GPU, si disponible). |
''
|
output_dir |
str | None | optional
|
Répertoire où enregistrer les résultats annotés. Par défaut, il s'agit d'un dossier "labels" dans le même répertoire que "data". |
None
|