Постановка задачи

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

Описание модели

За основу была взята модель ResNet18 с 18 слоями, а потом, чтобы подстроить под необходимую задачу, последний слой был заменен на линейный, с одним выходом, чтобы была возможность получить необходимый результат. Тренировка была с размером батча 64 и 10 эпохами, проводилась на основе датасета с 5000 разными лицами.

Результаты

Удалось добиться ошибки на тестовой выборки примерно 0.08 mse ~ в среднем 0.29 если извлечь корень, что довольно хороший результат для такой задачи. А также добавили функцию определения красоты в реальном времени