This project detects forest fires from images using Convolutional Neural Networks (CNN). It is a deep learning-based image classification system that distinguishes between fire and non-fire images, supporting early detection of wildfires to reduce damage and protect the environment.
- Goal: Binary classification of images into
FireandNo Fire. - Dataset: Collected from Kaggle β Wildfire Image Dataset.
- Framework: TensorFlow/Keras for building and training the CNN model.
- Application: Can be used in real-time monitoring systems for fire-prone regions.
| Task | Method |
|---|---|
| Processing visual data | Convolutional Neural Networks (CNN) |
| Image classification | Supervised learning (Binary classification) |
| Vision-based AI detection | Roboflow + YOLO (alternative/extension) |
π§ Note: While YOLO is often used for object detection, this project focuses on image classification using CNN. YOLO integration can be done in future versions for real-time video surveillance.
- Python
- TensorFlow / Keras
- OpenCV
- NumPy / Pandas
- Matplotlib / Seaborn
- Roboflow (for dataset visualization or deployment, optional)
-
Data Collection & Loading
- Dataset from Kaggle: fire and non-fire images.
- Binary classification task.
- Ensure all images are resized to the same dimensions (e.g., 224x224).
-
Image Preprocessing & Augmentation
- Resize, normalize, and convert images to arrays.
- Augment data (flip, rotate, zoom) using
ImageDataGenerator.
-
Model Building
- Construct a CNN using TensorFlow/Keras.
- Input layer β Convolutional layers β MaxPooling β Dense layers β Output layer with sigmoid.
-
Model Training
- Train the model on the preprocessed dataset.
- Use validation split for performance monitoring.
- Evaluate accuracy and loss using graphs.
-
Testing & Deployment
- Test model on unseen images.
- Optional: Deploy using Flask or Streamlit.