Репозиторий для обучения нейросетевых моделей по семантической сегментации + пример использования моделей на примере веб-приложения
Ноутбук для обучения сети Unet - baseline-train.ipynb
Код написан под формат данных CamVid из CVAT
Код веб-сервиса по изменению цвета волос и кожи, основанный на работе нейронной сети из примера - web.py. В примере рассматривается сеть, которая сегментирует кожу и волосы на фотографиях. Cсылка на сайт - веб-приложение
camvid-dataset из видео (трехклассовая сегментация) доступен по этой ссылке - DATASET
Необходимо иметь установленный python 3 любой версии.
Данные команды требуется запускать последовательно в терминале:
- Склонируйте к себе этот репозиторий
- Перейдите с помощью команды cd в созданную папку
- Загрузите все необходимые библиотеки:
PS: Лучше torch ставить сразу с поддержкой gpu если она имеется:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
либо если нет cuda:
pip install torch torchvision
далее запустить надо:
pip install -r requirements.txt
После этого можно работать с ноубуком обучения.
Но если есть желание запустить локально веб-сайт, то необходимо в терминате запустить эту команду:
streamlit run web.py
Имеется подробный туториал по работе с данным репозиторием, в котором рассказаны основные теоретические и практические моменты по обучению моделей семантической сегментации + использования моделей на практике
YouTube видео доступно по ССЫЛКЕ