/Fire-Detection

Primary LanguageJupyter Notebook

Fire-Detection

Fire and smoke detection using pytorch.

Download the Dataset - download

Dataset Folder -

Train
    - Fire
    - Neutral
    - Smoke       
Test
    - Fire
    - Neutral
    - Smoke
    
Dataset contains 1000 images of each class

Model Structure -

I have used trasfer learning using ResNet50 pretained model.

Upcoming Work -

  1. Fine-tuning of Training model.
  2. Inference Model: Implementation of OpenCV for real-time detection.