File size: 2,121 Bytes
7c23455 | 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 | ---
library_name: keras
pipeline_tag: image-classification
tags:
- keras
- tensorflow
- computer-vision
- image-classification
- fire-detection
- wildfire-detection
- binary-classification
- transfer-learning
---
# Fire Detection Models
A collection of trained deep learning models for binary fire detection.
These models classify an input image into one of two classes:
- **Fire**
- **No Fire**
The repository contains multiple CNN architectures trained as part of a broader deep learning project focused on fire detection and wildfire-related computer vision.
## Models
| Model | Architecture | Task |
|---|---|---|
| `Custom_CNN_binary_classification.keras` | Custom CNN | Fire / No Fire |
| `efficientnetb0_binary_classification.keras` | EfficientNetB0 | Fire / No Fire |
| `resnet50_binary_classification.keras` | ResNet50 | Fire / No Fire |
| `vgg16_binary_classification_finalized.keras` | VGG16 | Fire / No Fire |
## Model Format
All models are saved using the Keras `.keras` format.
```python
import tensorflow as tf
model = tf.keras.models.load_model(
"efficientnetb0_binary_classification.keras"
)
model.summary()
```
Replace the filename with the model you want to load.
## Intended Use
These models are intended for:
- research
- experimentation
- educational purposes
- model comparison
- further fine-tuning
- development of fire-detection applications
## Limitations
Model performance depends on the dataset, preprocessing pipeline, image quality, and deployment environment.
The models should be independently evaluated on an appropriate test dataset before being used in real-world or safety-critical applications.
## Project Context
These models are part of a larger deep learning project containing:
- fire detection
- fire severity classification
- feature extraction
- dimensionality reduction
- clustering
- recommendation generation
- supporting datasets and notebooks
## 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.
|