Wildfire detection with neural networks

Quickstart

1. Download the repository and pretrained weights

git clone git@github.com:tomek-l/fire-detect-nn.git
cd fire-detect-nn
wget https://dl.dropbox.com/s/6t17srif65vzqfn/firedetect-densenet121-pretrained.pt --directory-prefix=weights/

2. Inference on pretrained weights

pip3 install -r requirements.txt
python3 inference.py 

3. Training & Testing

• download the dataset.
• unzip the folder to data/fire-detect-nn-public-combo

python3 train.py # training 
python3 train-with-gradcam.py # for the model with gradcam (for the heatmap output)
python3 test.py # evaluation

For the GradCAM heatmap generation use jupyter notebook and navigate to inference-video.ipynb