YOLO Mask Detection System

About

A system capable of detecting masks developed using YOLOv3 Darknet. This system can detect mask in image, video or in live webcam.

Dataset Link:

Kaggle Dataset

Model Training File:

yolov3-mask-detection-final1.ipynb

How to run?

1. Install Python 3.8.6 from Python3.8

2. Install all requirements using:

pip install -r requirements.txt 

Better make a virtual environment before installing.

3. Download the weight from this link: YOLO Weights

4. To detect in image:

py test_on_photo.py

5. To detect in live video:

py test_on_live_video.py

Outputs

Detections in Image

Detections in Video

Detection Video

output_on_video.mp4

Detections in Live Stream from Webcam

output_on_live_webcam.mp4

P.S: FPS was a bit low so detections were lagging in live video stream

Acknowledgement

Special Thanks to Siddhant Sharma, Abhinandan Shrestha, Nitess Kumar Das