An open API service indexing awesome lists of open source software.

https://github.com/olafenwamoses/firenet

A deep learning model for detecting fire in video and camera streams
https://github.com/olafenwamoses/firenet

Last synced: 6 months ago
JSON representation

A deep learning model for detecting fire in video and camera streams

Awesome Lists containing this project

README

          

# FireNet
FireNet is an artificial intelligence project for real-time fire detection.






FireNet is a real-time fire detection project containing an annotated dataset, pre-trained models and inference codes, all created to ensure that machine learning systems can be trained
to detect fires instantly and eliminate false alerts. This is part of DeepQuest AI's to train machine learning systems to
perceive, understand and act accordingly in solving problems in any environment they are deployed.

This is the first release of the FireNet. It contains an annotated dataset of 502 images splitted into 412 images for training and 90 images for validation.


>>> DOWNLOAD, TRAINING AND DETECTION:


The FireNet dataset is provided for download in the release section of this repository.
You can download the dataset via the link below.

https://github.com/OlafenwaMoses/FireNET/releases/download/v1.0/fire-dataset.zip

We have also provided a [ImageAI](https://github.com/OlafenwaMoses/ImageAI) codebase to train a YOLOv3 detection model on the images
and perform detection in mages and videos using a pre-trained model (also using YOLOv3) provided in the release section of this repository.
The python codebase is contained in the fire_net.py file and the detection configuration JSON file for detection is also provided the
detection_config.json. The pretrained YOLOv3 model is available for download via the link below.


https://github.com/OlafenwaMoses/FireNET/releases/download/v1.0/detection_model-ex-33--loss-4.97.h5





Running the experiment or detection requires that you have **Tensorflow**, and **Keras**, **OpenCV** and **ImageAI** installed. You can install this dependencies via the commands below.


- Tensorflow 1.4.0 (and later versions) Install or install via pip

 pip3 install --upgrade tensorflow 


- OpenCV Install or install via pip
 pip3 install opencv-python 


- Keras 2.x Install or install via pip
 pip3 install keras 


- ImageAI 2.0.3
pip3 install imageai --upgrade 



>>> Video & Prediction Results


Click below to watch the video demonstration of the trained model at work.













References


1. Joseph Redmon and Ali Farhadi, YOLOv3: An Incremental Improvement

https://arxiv.org/abs/1804.02767