/yolo_fire_detection

People Counter and Fire Detection using YOLOv3

Primary LanguageJupyter Notebook

SCDFXIBM Fire Detection Algorithm

People Counter and Fire Detection using YOLOv3

click below to watch a demo of this system

IBM Cloud

Deployment

1. Download the weights to .\Data\Model_Weights

2. Run the Model on Input Video:

  • Yolov3 Dependencies should be in the same folder as the notebook before running
  • The Following Line in the notebook will process videos ,

drawing

- Drag your own video to be analysed to the same folder as this notebook (here its "cctv_4.mp4") under "video_path" - Set your output directory under "output_path" - Once the video has been processed, a labeled video should be created in the output path

3. Create Analytical Density Plot :

  • 2 Arrays (People Counter, Fire Counter) will be returned by the previous cell
  • In this Demonstration, 4 CCTV footage has already been analysed and stored in a .npy array

drawing

- The Following Code will then output a graphical report of the detections:

drawing

4. Upload to IBM Cloud

  • The following code retrieves images from IBM cloud
  • This code then runs the yolov3 model and outputs a labeled image
  • This labeled image is then uploaded back to IBM Cloud

drawing

Note:

  • Technical limitations: This model can ideally run on ibm watson (.h5 file can be uploaded). However, there are many dependency issues that would need to be solved (tensorflow version mismatch, figuring out access to .py scripts that contains dependencies). For the purpose of this Competition, this algorithm will be run locally. Only IBM cloud is used for storage.