Skip to content

prag-matic/wild_fire_detection_ai_modal_training

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Wildfire Prevention in Protected Forests Using Neural Networks

Project Overview

Wildfires are among the most destructive natural disasters, causing severe damage to biodiversity, wildlife habitats, and air quality. This project utilizes Deep Learning and Computer Vision to develop an automated early detection system.

By analyzing aerial and ground-level forest images, the system predicts wildfire risks and detects early signs of fire/smoke, aiming to minimize false alarms and improve response times in protected forest environments.

Key Features

  • Multi-Model Analysis: A comparative study of 6 different neural network architectures.
  • Robust Preprocessing: Implementation of ImageNet normalization, resizing to $224\times224$, and extensive data augmentation (flips, rotations).
  • Bias Mitigation: Stratified sampling and balanced datasets to ensure ethical modeling and fairness.
  • High Performance: The final model achieves a 99.41% F1-Score, prioritizing recall to ensure safety.

Dataset

The model was trained on a combined dataset to ensure diversity and generalization:

  1. Primary Dataset: 2,700 high-resolution RGB images (Wildfire Dataset). https://www.kaggle.com/datasets/elmadafri/the-wildfire-dataset
  2. Supplemental Dataset: 5,050 additional images (Forest Fire Images) to reduce overfitting. https://www.kaggle.com/datasets/mohnishsaiprasad/forest-fire-images/data
  3. Split: Data was divided into Training (70%), Validation (15%), and Test (15%) sets.

Model Architectures

The project implemented and evaluated the following architectures:

Baseline Models:

  • Multi-Layer Perceptron (MLP): A custom fully connected network built from scratch.
  • AlexNet: A standard 8-layer CNN implemented in PyTorch.

Transfer Learning Models (Fine-Tuned):

  • VGG16: Fine-tuned for binary classification.
  • ResNet50: Utilized for capturing deep, complex patterns.
  • MobileNetV2: Tested for efficiency in resource-constrained environments.
  • InceptionV3: The best-performing model, utilizing multi-scale convolution modules.

📊 Results & Comparative Analysis

The models were evaluated based on Accuracy, Precision, Recall, and F1-Score. The InceptionV3 (Fine-Tuned with SGD) was selected as the final model due to its superior balance of precision and recall.

Model Accuracy Precision Recall F1 Score
InceptionV3 (SGD) 🏆 0.9941 0.9909 0.9974 0.9941
VGG16 (Adam) 0.9862 0.9768 0.9961 0.9863
MobileNetV2 (Deep FT) 0.9743 0.9652 0.9842 0.9746
ResNet50 (SGD) 0.9645 0.9391 0.9934 0.9655
AlexNet (Adam) 0.9533 0.9186 0.9947 0.9551
MLP (Wide) 0.9322 0.9317 0.9329 0.9323

Data derived from the Comparative Analysis Table.

InceptionV3 Predictions

InceptionV3 _results

Technologies Used

  • Language: Python
  • Frameworks: PyTorch, NumPy
  • Visualization: Matplotlib
  • Environment: Jupyter Notebook / Google Colab

Author

Perera B.P.N.

  • Student ID: IT24100327
  • Faculty of Computing, SLIIT UNI

License

This project is licensed under the MIT License.

About

An AI-based system for early wildfire detection in protected forests. Features a comparative study of 6 architectures (MLP, AlexNet, VGG16, ResNet50, MobileNetV2, InceptionV3), achieving 99.4% F1-score with a fine-tuned InceptionV3 model.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors