superflash41's picture
Update README.md
5b60c25 verified
|
raw
history blame
549 Bytes

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:
    import tensorflow as tf
    
    model = tf.keras.models.load_model("ensemble_model.keras")
    prediction = model.predict(...)
    

---
license: mit
---