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