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
---