672 B
672 B
Face Mask Detection
Clone the repository and run project (it uses saved model 'face_mask_detection2.hp') or:
- Download dataset from: https://www.kaggle.com/andrewmvd/face-mask-detection
- In the project create folder 'data'
- Put downloaded 'annotations' and 'images' folders into 'data' folder
- Train model yourself:
- execute jupyter notebook 'data_preprocessing.ipynb' - data preprocessing
- execute jupyter notebook 'train_model.ipynb' - it will overwrite face_mask_detection2.hp' file
- run project
To run project:
pip3 install virtualenv
virtualenv venv
source venv/bin/activate
python3 -m pip install -r requirements.txt
python3 detect_mask.py