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