Fire and Gun Detection

How to use yolo.py:

usage: yolo.py [-h] [--webcam WEBCAM] [--playvideo PLAYVIDEO]
               [--image IMAGE] [--videopath VIDEOPATH]
               [--imagepath IMAGEPATH] [--verbose VERBOSE]
optional arguments:   -h, --help            show this help message and exit   --webcam WEBCAM       True/False   --playvideo PLAYVIDEO                         Tue/False   --image IMAGE         Tue/False   --videopath VIDEOPATH                         Path of video file   --imagepath IMAGEPATH                         Path of image to detect objects   --verbose VERBOSE     To print statements

Weights File Backup

If the GitLFS file is not accessible - Download Weights and keep inside the project folder.

Move inside the project folder and use the following command:

python yolo.py --playvideo True --videopath videos/fire1.mp4

Dataset

Training done on google collab - Jupyter notebook

Demo: Youtube

Paper