I have created a Convolutional Neural Network from scratch to detect if a person is wearing face mask or not using PyTorch. To test my project, please extract the dataset given as zip file and then run video.py file.
Youtube Link for example video: https://youtu.be/7sggQ3YzmBM
- Numpy
- PyTorch (https://pytorch.org/)
- OpenCV (For Live detection using Webcam)
- scikit-image (For Image Augmentation)
- scikit-learn (For Accuracy metrics)
- Clone this repository.
- Install all the dependencies mentioned in requirements.txt
pip install -r requirements.txt
orpip3 install -r requirements.txt
- Install PyTorch using this guide https://pytorch.org/
- Run
video.py
file.