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.
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.
- Downloads last month
- 59