Мы создаем хостинговую платформу для исполнения моделей компьютерного зрения на любом устройстве. Модели исполняются на наших серверах и доступны по REST API
Так как проект был рассчитан только на мужчин, нужно было научить нейросеть определять пол.
Для этого мы взяли большой набор данных, состоящий из фотографий людей с указанием пола.
Для увеличения точности мы использовали нейросеть dlib, которая детектировала на фото лицо.
анализ одежды
02
Для анализа одежды мы также использовали модель сегментации, то есть выделения одежды, а дальше присваивали ей один из пяти классов:
Кроме этого нужно было научить нейросеть находить обнаженные фото, т.е. отсутствие одежды на фото.
casual
спортивная одежда
военная форма
деловая одежда
экстремальная экипировка
анализ фигуры
03
Для того, чтобы определять типаж человека на фото более точно, нужно было обучить нейросеть распознавать фигуру.
Мы решили взять два основных параметра - научиться понимать комплекцию человека и его примерный вес.
Так мы смогли отделять четыре класса:
худой
полный
нормальный
мускулистый
анализ зон растительности на лице и цвета волос
04
Для начала нужно было определить тип растительности по зонам или ее отсутствие, а дальше определить цвет волос/бороды.
Для этого нам также понадобился детектор лица, а следующим шагом мы использовали предобученную модель Tiramisu и набор правил, чтобы детекция была более точной.
После этого нужно было определить цвет волос:
анализ фона
05
Последней задачей была классификация фона — нужно было попробовать определить место съемки или характерные приметы, такие как море, машина и так далее.
Из-за низкого качества реальных данных результат получался не очень точным, поэтому в итоговой модели практически не учитывался.
Еще более подробную презентацию проекта можно посмотреть здесь
«Как ни странно, трудностей почти не было. Сложнее всего было классифицировать фон. Мы очень долго с ним возились, но в результате просто решили убрать – оказалось, что остальных параметров достаточно».
Результаты проекта
10/
Было интересно и драйвово, потому что многие вещи были в новинку: