Ссылка для ultralytics/data/annotator.py
Примечание
Этот файл доступен по адресу https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/annotator .py. Если ты заметил проблему, пожалуйста, помоги исправить ее, отправив Pull Request 🛠️. Спасибо 🙏!
ultralytics.data.annotator.auto_annotate(data, det_model='yolov8x.pt', sam_model='sam_b.pt', device='', output_dir=None)
Автоматически аннотируй изображения, используя модель обнаружения объектов YOLO и модель сегментации SAM .
Параметры:
Имя | Тип | Описание | По умолчанию |
---|---|---|---|
data |
str
|
Путь к папке, содержащей изображения, которые нужно аннотировать. |
требуется |
det_model |
str
|
Предварительно обученная модель обнаружения YOLO . По умолчанию - 'yolov8x.pt'. |
'yolov8x.pt'
|
sam_model |
str
|
Предварительно обученная модель сегментации SAM . По умолчанию - 'sam_b.pt'. |
'sam_b.pt'
|
device |
str
|
Устройство, на котором будут запускаться модели. По умолчанию это пустая строка (CPU или GPU, если есть). |
''
|
output_dir |
str | None | optional
|
Каталог для сохранения аннотированных результатов. По умолчанию это папка 'labels' в той же директории, что и 'data'. |
None
|