Referência para ultralytics/data/annotator.py
Nota
Este ficheiro está disponível em https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/annotator .py. Se detectares um problema, por favor ajuda a corrigi-lo contribuindo com um Pull Request 🛠️. Obrigado 🙏!
ultralytics.data.annotator.auto_annotate(data, det_model='yolov8x.pt', sam_model='sam_b.pt', device='', output_dir=None)
Anota automaticamente imagens utilizando um modelo de deteção de objectos YOLO e um modelo de segmentação SAM .
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
data |
str
|
Caminho para uma pasta que contém imagens a serem anotadas. |
necessário |
det_model |
str
|
Modelo de deteção YOLO pré-treinado. Usa como predefinição 'yolov8x.pt'. |
'yolov8x.pt'
|
sam_model |
str
|
Modelo de segmentação SAM pré-treinado. Usa como predefinição 'sam_b.pt'. |
'sam_b.pt'
|
device |
str
|
Dispositivo para executar os modelos. Usa como padrão uma string vazia (CPU ou GPU, se disponível). |
''
|
output_dir |
str | None | optional
|
Diretório para guardar os resultados anotados. A predefinição é uma pasta 'labels' no mesmo diretório que 'data'. |
None
|