Набор утилит для проетков глубокого обучения в задачах технического зрения.
Файл | Краткое описание |
---|---|
utils.py |
Утилиты общего назначения. |
cv_utils.py |
Утилиты для обработки изображений. |
cvat.py |
Работа с CVAT-датасетами. Позволяет читать и интерпретировать CVAT-бекапы. |
yolo.py |
Работа с YOLO-датасетами. Используется для преобразование заргуженных с помощью cvat.py задач в YOLO-подвыборку.``. |
samal.py |
Работа с моделью Segment Anything (SAM). Используется для предварительной разметки изображений в датасетах.``. |
show.sh |
Мониторинг процесса. Выводит последние строки заданного файла (по-умолчанию - ./nohup.out ), обновляя вывод раз в несколько секунд. Позволяет отслеживать выводы процессов, запущенных через nohup. |
Команда | Краткое описание |
---|---|
conda install -y -c conda-forge lap |
Можно не устанавливать, если нет необходимости в использовании YOLO из ultralytics в режиме трекера. |
pip install ultralytics |
YOLO, SAM и ещё некоторые полезные фреймворки на PyTorch в одном модуле. |
pip install git+https://github.com/facebookresearch/segment-anything.git |
Segment Anything Model (SAM) |