Sign inSign up

kshitizb/pyrodetect

By kshitizb

•Updated over 1 year ago

AI-Powered Wildfire Detection

Image
Machine learning & AI
0

937

kshitizb/pyrodetect repository overview

ā šŸ”„ PyroDetect - AI-Powered Wildfire Detection

PyroDetect is an advanced wildfire detection system using a custom Convolutional Neural Network (CNN). This AI-driven model enhances fire detection accuracy while minimizing false alarms, making it ideal for real-time deployment.

ā šŸ”¹ Features

āœ… Custom CNN architecture for precise wildfire detection
āœ… Data augmentation techniques for robust performance
āœ… Optimized training with Adam optimizer & batch normalization
āœ… High accuracy and reduced false positives
āœ… Fully containerized for seamless deployment

ā šŸ“Œ Installation & Usage

⁠Run the Container
docker run -p 8020:8020 kshitizb/pyrodetect

Then open: http://localhost:8020/⁠

⁠Pull the Image
docker pull kshitizb/pyrodetect

ā šŸ“Œ Input & Output

  • Input: High-resolution aerial/ground wildfire images.
  • Output: Predicted wildfire presence with confidence scores.

ā šŸ“Œ Model Performance

ModelAccuracyPrecisionRecall
Custom CNN84.15%87.80%86.06%
⁠Evaluation Metrics
  • Confusion Matrix: TP: 216, TN: 129, FP: 30, FN: 35
  • ROC Curve: AUC = 0.90
  • Precision-Recall Curve: AUC = 0.99

ā šŸ“Œ Dataset

PyroDetect is trained on 2,700 high-resolution images collected from various sources:

  • Government wildfire databases
  • Public repositories (Flickr, Unsplash)
  • Aerial and ground-based images
MetricValue
Total Images2,700
Avg Resolution4057Ɨ3155 px
Min Resolution153Ɨ206 px
Max Resolution19699Ɨ8974 px

ā šŸ”¹ Contributors

šŸ‘Øā€šŸ’» Kshitiz Bhargava - Linkedin⁠ Github⁠

šŸ‘Øā€šŸ’» Manvendra Singh - Linkedin⁠ Github⁠

šŸ‘Øā€šŸ’» Abeer Mathur - Linkedin⁠ Github⁠

Tag summary

Content type

Image

Digest

sha256:8478fd14f…

Size

501.7 MB

Last updated

over 1 year ago

docker pull kshitizb/pyrodetect