An End-to-end Campus Fire Detection Algorithm Incorporating Priori Features
To address the imbalance in detection speed and accuracy caused by smoke, flame, and background confusion in campus video surveillance scenarios, an end-to-end campus fire detection algorithm incorporating priori features is proposed. First, a feature pyramid fusion module guided by priori features is constructed to cascade the multi-scale feature fusion module. The priori features are individually pre-trained by Convolutional Neural Network (CNN) classification network. The feature complementarity between the two is utilized to improve the feature representation of the detection network. Second, a lightweight set prediction detection head is designed. The head does not require manually designed anchors and post-processing strategies, enabling end-to-end detection while balancing speed and accuracy. Finally, a joint loss function that integrates the losses of categorical and bipartite matching is proposed. Focus on the drawbacks of the slow convergence associated with bipartite matching, both classification information and bipartite bit matching are considered in the detection results. Experimental results on the fire detection dataset demonstrate that the proposed algorithm improved average accuracy by 7.8% while maintaining detection speed, effectively balancing speed and accuracy in fire detection.
Authors
- Jie Wang (ORCID: https://orcid.org/0009-0000-9753-5252)
- Xuan Liu (ORCID: https://orcid.org/0000-0003-3335-7308)
- Xinlin Xie (ORCID: https://orcid.org/0000-0002-1988-8535)
- Yu Liang
Institutions
- Taiyuan University of Science and Technology (CN)
Publication Details
- Published
- 2024-12-06
- DOI
- https://doi.org/10.1109/ricai64321.2024.10911502
- Primary Topic
- Fire Detection and Safety Systems
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China
- Research and Development