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

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.