Школьник из Москвы научил нейросеть «читать» эмоции по фото

Ученик ИT-класса школы №548 «Царицыно» Дмитрий Филиппов разработал нейросеть, которая определяет эмоции по фотографии – от радости до нейтрального состояния.

Модель обучена на базе PyTorch и датасета FER-2013, где собраны изображения лиц с разными эмоциями. Чтобы пользоваться было проще, Дмитрий сделал отдельное приложение: достаточно загрузить фото, и система покажет, какое настроение на нем распознано.

В процессе разработки пришлось решать типичные задачи для ИИ: работать с небольшими и не всегда четкими изображениями и выравнивать данные. Для этого автор применил методы нормализации и защиты от переобучения.

Дальше – больше: в планах улучшить точность, добавить анализ видео в реальном времени и адаптировать модель под разные условия съемки.

Источник: ТАСС

#IT_News #ИИ #Образование

Подписаться в MAX