Drones (Sep 2024)

Real-Time Fire Detection: Integrating Lightweight Deep Learning Models on Drones with Edge Computing

  • Md Fahim Shahoriar Titu,
  • Mahir Afser Pavel,
  • Goh Kah Ong Michael,
  • Hisham Babar,
  • Umama Aman,
  • Riasat Khan

DOI
https://doi.org/10.3390/drones8090483
Journal volume & issue
Vol. 8, no. 9
p. 483

Abstract

Read online

Fire accidents are life-threatening catastrophes leading to losses of life, financial damage, climate change, and ecological destruction. Promptly and efficiently detecting and extinguishing fires is essential to reduce the loss of lives and damage. This study uses drone, edge computing, and artificial intelligence (AI) techniques, presenting novel methods for real-time fire detection. This proposed work utilizes a comprehensive dataset of 7187 fire images and advanced deep learning models, e.g., Detection Transformer (DETR), Detectron2, You Only Look Once YOLOv8, and Autodistill-based knowledge distillation techniques to improve the model performance. The knowledge distillation approach has been implemented with the YOLOv8m (medium) as the teacher (base) model. The distilled (student) frameworks are developed employing the YOLOv8n (Nano) and DETR techniques. The YOLOv8n attains the best performance with 95.21% detection accuracy and 0.985 F1 score. A powerful hardware setup, including a Raspberry Pi 5 microcontroller, Pi camera module 3, and a DJI F450 custom-built drone, has been constructed. The distilled YOLOv8n model has been deployed in the proposed hardware setup for real-time fire identification. The YOLOv8n model achieves 89.23% accuracy and an approximate frame rate of 8 for the conducted live experiments. Integrating deep learning techniques with drone and edge devices demonstrates the proposed system’s effectiveness and potential for practical applications in fire hazard mitigation.

Keywords