Распознавание эмоций

Датасет был сгенерирован с использованием модели Stable Diffusion. Генерация изображений осуществлялась в google colab (ноутбук). Готовый датасет доступен здесь. Для нахождения ключевых точек использовалась библиотека Mediapipe (face landmarker model). Для классификации точек использовалась библиотека Scikit-learn.

Результаты

Использование

1. Создать виртуальную среду с python 3.11 и установить requirements.txt
2. Выполнить python inference_model.py. Можно дополнительно указать путь до папки или файла, которые нужно обработать. По умолчанию изображения и видео берутся из папки assets.
3. Результаты выполнения будут сохранены в папке output.
4. Для обработки с вебкамеры выполнить python webcam.py. Для выхода нажать q.

Ограничения

Для обучения использовался небольшой набор искусственно сгенерированных данных. Модель имеет сложности с распознаванием эмоций с данных, отличающихся от тренировочных.