О проекте

Мы создаём модуль 2D-виртуальной примерки одежды, основанный на моделях компьютерного зрения. Пользователь загружает фото, и алгоритм "примеряет" выбранную одежду, учитывая посадку и параметры тела.

Также мы развиваем систему рекомендаций по размеру и стилю.

Задачи

1. Разработка и обучение моделей для визуальной примерки одежды:

- GAN-модель (или диффузионная как один из путей для улучшения базового качества модели),

- проведение ресерча по существующим статьям в области, попытка реализации схожих архитектур моделей, их обучения (проведение экспериментов)

- регрессор/пре-процессор (из изображений путем применения вышеописанных методов извлекаются признаки, однако далее идет этап непосредственного предсказания параметров тела человека, чтобы осуществить в том числе 3D примерку),

- итеративное переобучение на датасетах.

2. Разработка модуля валидации фото пользователей (качество снимка, поза, освещение).

3. Сбор и подготовка данных: скрапинг, аугментация, ручная разметка (с целью создания синтетического датасета для обучения) или покупка датасетов.

4. Разработка рекомендательных сервисов:

соотнесение моделей по размерной сетке,

рекомендации по параметрам тела.

Это предполагается делать с помощью градиентного бустинга.

5. Совместный вывод ключевых модулей в продакшн (вместе с backend-разработчиками).

Требования

- Опыт в ML/Computer Vision, в том числе GAN (опыт обучения подобных сетей, будет плюсом наличие опыта с диффузионными моделями).

- Уверенное знание Python (PyTorch/TensorFlow).

- Умение проводить ресерч и ставить эксперименты (логгирование итогов каждого, тестирование своего кода, умение находить уязвимые места в пайплайне).

- Docker, базовая инфраструктура релевантная ML.

- Навыки работы с большими датасетами изображений.

- Понимание полного ML pipeline (от подготовки данных до inference в проде).

- Будет плюсом опыт выведения нейронных сетей в продакшен, а также инференс больших моделей на проде.

‍

По условиям: 12-15 часов в неделю, по деньгам договариваемся индивидуально
‍

Откликнуться: @forgtp154