Skip to content

Набор утилит для проектов глубокого обучения в задачах технического зрения.

License

Notifications You must be signed in to change notification settings

NikitaShubin/dl_utils

Repository files navigation

DL Utils

Набор утилит для проетков глубокого обучения в задачах технического зрения.

Основной состав:

Файл Краткое описание
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)

About

Набор утилит для проектов глубокого обучения в задачах технического зрения.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published