Определение наличия маски на лице

Цель проекта

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

Инструменты

       

Данные

Для проекта был использован набор данных "Face Mask Detection ~12K Images Dataset" Kaggle.
Характеристики данных: 

• Формат изображений: RGB 
• Размеры изображений: от 25x25 до 563x563 пикселей

Структура модели

Обучение модели и оценка

Обучение модели:

• Модель была обучена на тренировочном наборе данных с использованием алгоритма оптимизации Adam и функции потерь бинарной кросс-энтропии.
• Для предотвращения переобучения была использована ранняя остановка (early stopping).
• Обучение проводилось в течение 5 эпох.

Оценка на тестовом наборе данных:

• После завершения обучения модель была протестирована на тестовом наборе данных для оценки ее производительности.
• Полученная точность на тестовом наборе данных составила 95.56%.