A High-Precision Ensemble Model for Forest Fire Detection in Large and Small Targets

Abstract

Forest fires are major forestry disasters that cause loss of forest resources, forest ecosystem safety, and personal injury. It is often difficult for current forest fire detection models to achieve high detection accuracy on both large and small targets at the same time. In addition, most of the existing forest fire detection models are single detection models, and using only a single model for fire detection in a complex forest environment has a high misclassification rate, and the accuracy rate needs to be improved. Aiming at the above problems, this paper designs two forest fire detection models (named WSB and WSS) and proposes an integrated learning-based forest fire detection model (named WSB_WSS), which also obtains high accuracy in the detection of forest fires with large and small targets. In order to help the model predict the location and size of forest fire targets more accurately, a new edge loss function, Wise-Faster Intersection over Union (WFIoU), is designed in this paper, which effectively improves the performance of the forest fire detection algorithm. The WSB model introduces the Simple-Attention-Module (SimAM) attention mechanism to make the image feature extraction more accurate and introduces the bi-directional connectivity and cross-layer feature fusion to enhance the information mobility and feature expression ability of the feature pyramid network. The WSS model introduces the Squeeze-and-Excitation Networks (SE) attention mechanism so that the model can pay more attention to the most informative forest fire features and suppress unimportant features, and proposes Spatial Pyramid Pooling-Fast Cross Stage Partial Networks (SPPFCSPC) to enable the network to extract features better and speed up the operation of the model. The experimental findings demonstrate that the WSB model outperforms other approaches in the context of identifying forest fires characterized by small-scale targets, achieving a commendable accuracy rate of 82.4%, while the WSS model obtains a higher accuracy of 92.8% in the identification of large target forest fires. Therefore, in this paper, a more efficient forest fire detection model, WSB_WSS, is proposed by integrating the two models through the method of Weighted Boxes Fusion (WBF), and the accuracy of detecting forest fires characterized by small-scale targets attains 83.3%, while for forest fires with larger dimensions, the accuracy reaches an impressive 93.5%. This outcome effectively leverages the strengths inherent in both models, consequently achieving the dual objective of high-precision detection for both small and large target forest fires concurrently.