- 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
![@dependabot[bot]](https://avatars.githubusercontent.com/in/29110?s=64&v=4){kind=small}