Downloads · 30 days
7
1% of all-time downloads
vit-base-patch16-224-in21k-fire-detection
Gurveer05/vit-base-patch16-224-in21k-fire-detection
vit-base-patch16-224-in21k-fire-detection is a image classification model from Gurveer05. Use it when you need a label for an image. It is set up for transformers.
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.
All-time downloads
824
Public
Parameters
85.8M
16.5 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors343 MB · 100%
At a glance
- Task
- Image Classification
- Library
- transformers
- Model type
- vit
- Access
- Public
- Created
- Mar 20, 2024
- Updated
- Apr 3, 2024
- SHA
- 1b9da2ba
Model card
From the Hugging Face model README
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(Dataset in input)Github Repository with website integration
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- Created
- Mar 20, 2024
- Updated
- Apr 3, 2024