superflash41 commited on
Commit
5b60c25
·
verified ·
1 Parent(s): 7299ad4

Update README.md

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