FCMI-YOLO: An efficient deep learning-based algorithm for real-time fire detection on edge devices.

Journal: PloS one
Published Date:

Abstract

The rapid development of Internet of Things (IoT) technology and deep learning has propelled the deployment of vision-based fire detection algorithms on edge devices, significantly exacerbating the trade-off between accuracy and inference speed under hardware resource constraints. To address this issue, this paper proposes FCMI-YOLO, a real-time fire detection algorithm optimized for edge devices. Firstly, the FasterNext module is proposed to reduce computational cost and enhance detection precision through lightweight design. Secondly, the Cross-Scale Feature Fusion Module (CCFM) and the Mixed Local Channel Attention (MLCA) mechanism are incorporated into the neck network to improve detection performance for small fire targets and reduce resource consumption. Finally, the Inner-DIoU loss function is proposed to optimize bounding box regression. Experimental results on a custom fire dataset demonstrate that FCMI-YOLO increases mAP@50 by 1.5%, reduces parameters by 40%, and lowers GFLOPs to 28.9% of YOLOv5s, demonstrating its practical value for real-time fire detection in edge scenarios with limited computational resources. The core code and dataset are available at https://github.com/ JunJieLu20230823/code.git.

Authors

  • Junjie Lu
    Xiangyang Central Hospital, Affiliated Hospital of Hubei University of Arts and Science, Hubei Province, Xiangyang, 441021, China.
  • Yuchen Zheng
    Medical College, Guizhou University, Guiyang, 550025, Guizhou, China.
  • Liwei Guan
    College of Physics and Energy, Fujian Normal University, Fujian, China.
  • Bing Lin
    Affiliated Hospital of Chengdu University of Traditional Chinese Medicine, No. 37 Shierqiao Avenue, Jinniu District, Chengdu 610075, China.
  • Wenzao Shi
    College of Photonic and Electronic Engineering, Fujian Normal University, Fujian, China.
  • Junyan Zhang
  • Yunping Wu
    College of Photonic and Electronic Engineering, Fujian Normal University, Fujian, China.