AbdullahImran commited on
Commit
2ab3b70
·
verified ·
1 Parent(s): d15fcf3

Add fire severity model card

Browse files
Files changed (1) hide show
  1. README.md +95 -0
README.md ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: keras
3
+ pipeline_tag: image-classification
4
+ tags:
5
+ - keras
6
+ - tensorflow
7
+ - computer-vision
8
+ - image-classification
9
+ - fire-detection
10
+ - fire-severity
11
+ - severity-classification
12
+ - multi-class-classification
13
+ - transfer-learning
14
+ ---
15
+
16
+ # Fire Severity Classification Models
17
+
18
+ A collection of trained deep learning models for three-class fire severity classification.
19
+
20
+ The models classify fire images into three severity categories:
21
+
22
+ - **Mild**
23
+ - **Moderate**
24
+ - **Severe**
25
+
26
+ Multiple CNN architectures are included for model comparison.
27
+
28
+ ## Models
29
+
30
+ | Model | Architecture | Task |
31
+ |---|---|---|
32
+ | `efficientnetb0_tri_classification.keras` | EfficientNetB0 | Mild / Moderate / Severe |
33
+ | `resnet50_tri_classification.keras` | ResNet50 | Mild / Moderate / Severe |
34
+ | `xception_tri_classification_finalized.keras` | Xception | Mild / Moderate / Severe |
35
+
36
+ ## Model Format
37
+
38
+ All models are saved using the Keras `.keras` format.
39
+
40
+ ```python
41
+ import tensorflow as tf
42
+
43
+ model = tf.keras.models.load_model(
44
+ "efficientnetb0_tri_classification.keras"
45
+ )
46
+
47
+ model.summary()
48
+ ```
49
+
50
+ Replace the filename with the model you want to load.
51
+
52
+ ## Classification Task
53
+
54
+ ```text
55
+ Input Image
56
+ |
57
+ v
58
+ CNN Model
59
+ |
60
+ v
61
+ Fire Severity
62
+ |
63
+ +---- Mild
64
+ |
65
+ +---- Moderate
66
+ |
67
+ +---- Severe
68
+ ```
69
+
70
+ ## Intended Use
71
+
72
+ These models are intended for:
73
+
74
+ - research
75
+ - experimentation
76
+ - educational purposes
77
+ - model comparison
78
+ - further fine-tuning
79
+ - fire severity analysis
80
+
81
+ ## Limitations
82
+
83
+ Model performance depends on the training dataset, preprocessing pipeline, image quality, class distribution, and deployment environment.
84
+
85
+ These models should be independently evaluated before being used in real-world or safety-critical applications.
86
+
87
+ ## Project Context
88
+
89
+ These models are part of a larger deep learning project involving fire detection, fire severity classification, feature extraction, dimensionality reduction, clustering, and recommendation generation.
90
+
91
+ ## License
92
+
93
+ No standardized open-source license has been specified for this repository.
94
+
95
+ Please refer to the original project and dataset terms before redistribution or commercial use.