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
- Christiano Machado (christianomachado10@gmail.com | github.com/chrismachado)