Detecting forest fires using deep learning.
| Model | Params (M) | IoU | F1 |
|---|---|---|---|
| deeplabv3_mobilenet_v3_large | 11.0 | 0.3544 | 0.5230 |
| deeplabv3_resnet101 | 60.9 | 0.375 | 0.5450 |
| deeplabv3_resnet50 | 42.0 | 0.3915 | 0.5623 |
| lraspp_mobilenet_v3 | 3.2 | 0.3713 | 0.5412 |
| segformer-b0 | 3.7 | 0.7446 | 0.8535 |
| segformer-b2 | 25.8 | 0.7671 | 0.8680 |
| ensemble | 0.3724 | 0.5423 |
Checkpoints of the trained models are available in releases.
segformer-b0 finetuned endpoint can also be found on huggingface.
- The notebooks will automatically download the self hosted dataset and checkpoints.
- The notebooks themselves are divided into subsections and are self-explanatory.