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
- Download the
.keras
file you want (e.g.ensemble_model.keras
). - Load it in Python:
import tensorflow as tf model = tf.keras.models.load_model("ensemble_model.keras") prediction = model.predict(...)
---
license: mit
---