Need help with FireNET?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

OlafenwaMoses
198 Stars 87 Forks MIT License 3 Commits 11 Opened issues

Description

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

Services available

!
?

Need anything else?

Contributors list

# 1,278
Python
pytorch
predict...
incepti...
3 commits

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 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 firenet.py file and the detection configuration JSON file for detection is also provided the <a href="detectionconfig.json" >detectionconfig.json. The pretrained YOLOv3 model is available for download via the link below.

<a href="https://github.com/OlafenwaMoses/FireNET/releases/download/v1.0/detectionmodel-ex-33--loss-4.97.h5" >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

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.