Skip to content

tomasz-lewicki/fire-detect-nn

Repository files navigation

Wildfire detection with neural networks

fire

Quickstart

  1. Download the repository and pretrained weights
git clone [email protected]: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