Skip to content

Commit

Permalink
update README
Browse files Browse the repository at this point in the history
  • Loading branch information
millionhz committed Apr 26, 2023
1 parent 6cf040f commit 0380e05
Showing 1 changed file with 13 additions and 4 deletions.
17 changes: 13 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,19 @@ Detecting forest fires using deep learning.

| Model | Params (M) | IoU | F1 |
| --- | --- | --- | --- |
| deeplabv3\_mobilenet\_v3\_large | 11.0 | 0.3544 | 0.523 |
| deeplabv3\_resnet101 | 60.9 | 0.375 | 0.545 |
| 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.868 |
| Ensemble Model | 26.5 | 0.0006 | 0.0012 |
| segformer-b2 | 25.8 | 0.7671 | 0.8680 |
| ensemble | | 0.3724 | 0.5423 |

## Checkpoints

Checkpoints of the trained models are available in releases.

## Installation and inference

- The notebooks will automatically download the self hosted dataset and checkpoints.
- The notebooks themselves are divided into subsections and are self-explanatory.

0 comments on commit 0380e05

Please sign in to comment.