Skip to content

tomasz-lewicki/fire-detect-nn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

b45d516 · Jun 22, 2022
Mar 6, 2021
Mar 9, 2021
Nov 2, 2020
Nov 2, 2020
Oct 28, 2020
Mar 6, 2021
Oct 28, 2020
Oct 28, 2020
Oct 28, 2020
Oct 6, 2020
Oct 28, 2020
Mar 10, 2021
Oct 28, 2020
Nov 2, 2020
Mar 12, 2021
Jun 22, 2022
Oct 21, 2020
Mar 6, 2021
Nov 2, 2020
Mar 12, 2021

Repository files navigation

Wildfire detection with neural networks

fire

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/
  1. Inference on pretrained weights
pip3 install -r requirements.txt
python3 inference.py 
  1. Training & Testing
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