Описание
Face Mask Detection
Этот проект представляет собой систему обнаружения наличия медицинских масок на лицах людей с использованием компьютерного зрения и глубокого обучения
Описание
Данная работа разработана в качестве учебного проекта для обнаружения, надета ли у человека медицинская маска. Для более серьезный условий, в частности где требуется соблюдение мер безопасности, таких как общественные места, медицинские учреждения и т.д. и там где система может быть интегрирована в видеокамеры или существующие системы безопасности для автоматического мониторинга и контроля потребуется более глубокое обучение модели.
Функции.
Обнаружение лиц на изображениях с использованием каскадов Хаара. Классификация лиц на наличие маски с помощью глубокой нейронной сети. Визуализация результатов обнаружения масок на лицах. Тренировка модели на собственном наборе данных для улучшения качества обнаружения. Dataset с изображениями находиться в папке проекта.
Принцип работы
Модель получает данные, обучается и на основе обучения считывает сначала для тестирования данные фотографии в папках и где вы можете увидеть в режиме онлайн прогресс распознавания. После открывается камера и вы наглядно проверите работу на себе.
Установка
1.Перед началом работы создайте папку dataset в корне проекта.
Создайте в этой папке две папки : with_mask и without_mask и загрузите туда соответствующие фотографии лиц людей в маске и без.
2.Установите необходимые зависимости.
-opencv-python==4.5.3
-numpy==1.21.2
-tensorflow==2.6.0
-scikit-learn==0.24.2
Использование
1.Запустите приложение, чтобы обучить модель:
python main.py train
2.Запустите приложение для тестирования модели:
python main.py test
3.Запустите камеру для обнаружения масок в реальном времени:
python main.py camera
Вклад
Проект открыт для вклада и улучшения. Не стесняйтесь отправлять pull request с предложенными изменениями.