Skip to content

Latest commit

 

History

History
50 lines (35 loc) · 1.54 KB

readme.md

File metadata and controls

50 lines (35 loc) · 1.54 KB
Raw

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.

  1. Download the weight from this link: YOLO Weights

  2. To detect in image:

py test_on_photo.py
  1. To detect in live video:
py test_on_live_video.py

Outputs

Detections in Image

Output1 Output2

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