FasterGDSF-DETR: A Faster End-to-End Real-Time Fire Detection Model via the Gather-and-Distribute Mechanism

Fire detection using deep learning has become a widely adopted approach. However, YOLO-based models often face performance limitations due to NMS, while DETR-based models struggle to meet real-time processing requirements. To address these challenges, we propose FasterGDSF-DETR, a novel fire detection model built upon the RT-DETR framework, designed to enhance both detection accuracy and efficiency. Firstly, this model introduces the FasterDBBNet backbone, which efficiently captures and retains feature information, accelerating the model’s convergence speed. Secondly, we propose the AIFI-GDSF hybrid encoder to reduce information loss in intra-scale interactions and improve the capability of detecting varying morphological flames. Furthermore, to better adapt to complex fire scenarios, we expand the dataset based on the KMU Fire and Smoke database and incorporate WIoU as the loss function to improve model robustness. Experimental results demonstrate that our proposed model surpasses mainstream object detection models in both accuracy and computational efficiency. FasterGDSF-DETR achieves a mean Average Precision of 71.5% on the self-constructed dataset, outperforming the YOLOv9 model of the same scale by 2.4 percentage points. This study introduces a novel task-specific enhancement to the RT-DETR framework, offering valuable insights for future advancements in fire detection technology.