В этом репозитории вы найдете код обучения нейронной сети, предсказывающей координаты 971 точек лица.
Сеть обучалась на основе датасета из 393 930 фотографий.
Пример работы нейронной сети:
- Были подготовлены функции для обработки изображений
- Далее были подготовлены функции для обучения, валидации и предсказания результатов
- Читаем и трансформируем данные для обучения и валидации. Размер батча - 256
- Определяем модель (предобученная resnet50), функцию потерь, метод оптимизации. В качестве функции потерь лучше всего себя показала smooth_l1_loss, также была опробирована WingLoss
- Добавляем дополнительный линейный слой над предобученной моделью
- Обучение и валидация модели
- Генерируем предсказание для загрузки