File size: 549 Bytes
5b60c25 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
# Fire Classification Models
This repository contains several Keras models trained for fire vs. no-fire image classification:
- **DenseNet**: `densenet_final.keras`
- **ResNet**: `resnet_final.keras`
- **Xception**: `xception_final.keras`
- **Ensemble**: `ensemble_model.keras`
## Usage
1. Download the `.keras` file you want (e.g. `ensemble_model.keras`).
2. Load it in Python:
```python
import tensorflow as tf
model = tf.keras.models.load_model("ensemble_model.keras")
prediction = model.predict(...)
```
---
license: mit
---
|