Fire Detection Based on Improved Feature Fusion

Fire is a serious disaster that endangers life and property worldwide. Accurate and prompt fire detection can provide early warning information for firefighting efforts. Deep learning-based object detection algorithms have proven to be useful in the field of fire detection. However, fire detection is a particularly challenging task, due to the low-resolution of fire images and the presence of blurring and deformation in the targets. Existing fire detection methods are limited by poor detection accuracy and slow speed in complex fire scenarios. To address these challenges, we propose a fire detection algorithm based on improved YOLOv5s. Firstly, the attention module is embedded into the network structure. We introduce this attention module into the last layer of the backbone to enhance feature extraction ability. Secondly, we modify the Neck structure to fully fuse different feature maps. Through fusing extra low-level features from the backbone, it utilizes various receptive fields to enhance the expression of feature. Thirdly, SIoU is introduced to achieve faster convergence during the training. SIoU take into account the direction of the mismatch to redefine penalty indicators. The experimental results show that our proposed algorithm outperforms commonly used models. Even compared to specific fire detectors, our model exhibits certain advantages.