File size: 1,023 Bytes
2616abe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
---
tags:
- image-classification
- wildfire-detection
datasets:
- your-dataset-name # Optional: Add the name of the dataset you used
---
# ResNet50 Fine-tuned for Wildfire Detection
This model is a ResNet50 fine-tuned on the [Wildfire Prediction Dataset](https://www.kaggle.com/datasets/abdelghaniaaba/wildfire-prediction-dataset) for image classification to detect wildfires.
## Model Details
The model is a ResNet50 architecture with a custom classification head. It was trained in two phases: first with the backbone frozen, and then with full fine-tuning.
## Training
The model was trained on the provided dataset with the following characteristics:
- **Train samples:** 30250
- **Validation samples:** 6300
- **Batch size:** 16
- **Epochs (Phase 1 - Head):** 5
- **Epochs (Phase 2 - Full Fine-tuning):** 3
## Performance
(Add details about the model's performance metrics, e.g., accuracy, loss, etc. from your training output)
## Usage
(Provide code examples on how to load and use the model for inference)
|