---
library_name: transformers
tags:
- climate
pipeline_tag: image-classification
---
The purpose of this model is to detect wildfires / forest fires in satellite imagery for early wildfire detection. It has been trained on a custom dataset from kaggle.

Outputs are encoded as follows:
- LABEL_0: Fire
- LABEL_1: No Fire

References:
- [`Kaggle Notebook`](https://www.kaggle.com/code/gurveersinghvirk/fire-detection-vit) (Dataset in input)
- [`Github Repository with website integration`](https://github.com/gurveervirk/wildfire-detection-in-satellite-image-ViT)