diff --git a/README.md b/README.md index b5af5d5..652508c 100644 --- a/README.md +++ b/README.md @@ -1,10 +1,11 @@ # Face Mask Detection -1. Clone repository -2. Download dataset from: https://www.kaggle.com/andrewmvd/face-mask-detection -3. In the project create folder 'data' -4. Put 'annotations' and 'images' folders into 'data' folder -5. You can now run project (it uses saved model 'face_mask_detection2.hp') or you can train model yoursel and to do that: +Clone the repository and run project (it uses saved model 'face_mask_detection2.hp') or: + +1. Download dataset from: https://www.kaggle.com/andrewmvd/face-mask-detection +2. In the project create folder 'data' +3. Put downloaded 'annotations' and 'images' folders into 'data' folder +4. 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