- 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/- Inference on pretrained weights
pip3 install -r requirements.txt
python3 inference.py
- 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 # evaluationFor the GradCAM heatmap generation use jupyter notebook and navigate to inference-video.ipynb