Первоначально предназначавшийся для решения задачи замены фона в сервисах видеоконференций датасет был дополнительно размечен и для задачи анализа лица. EasyPortrait состоит из 20 тыс. изображений 8,37 тыс. уникальных пользователей и разбит на тренировочную, валидационную и тестовую выборки. Каждому изображению соответствует сегментационная маска (двумерный массив в формате PNG), разделенная на девять классов, описывающих различные области лица.

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

EasyPortrait был представлен и выложен в общий доступ командой CV R&D SberDevices в мае 2023 года.

Препринт на arXiv

Репозиторий на GitHub

👤 Александр Капитанов