chrismachado / mask-detection-cnn

This project aims to create a neural network model that is able to identify individuals who are wearing a mask.

Face Mask Recognition

Usage

Cloning this repo using

git clone https://github.com/chrismachado/mask-detection-cnn

Install python dependencies with pip

pip install -r requirements.txt 

Then, if you just want to train and test the model, just type

python train_mask_detector.py -d dataset 

Or, you can run pre-trained model in output folder by using the following command

python detect_mask_image.py -i examples/example_01.png

For more information about the script, you can use help command

python detect_mask_image.py -h

Author

• Christiano Machado ([e-mail] | github.com/chrismachado)

References

• Prajna’s GitHub repository
• Adrian Rosebrock's Article