Wildfire Identification Based on an Improved MobileNetV3-Small Model

Abstract

In this paper, an improved MobileNetV3-Small algorithm model is proposed for the problem of poor real-time wildfire identification based on convolutional neural networks (CNNs). Firstly, a wildfire dataset is constructed and subsequently expanded through image enhancement techniques. Secondly, an efficient channel attention mechanism (ECA) is utilised instead of the Squeeze-and-Excitation (SE) module within the MobileNetV3-Small model to enhance the model’s identification speed. Lastly, a support vector machine (SVM) is employed to replace the classification layer of the MobileNetV3-Small model, with principal component analysis (PCA) applied before the SVM to reduce the dimensionality of the features, thereby enhancing the SVM’s identification efficiency. The experimental results demonstrate that the improved model achieves an accuracy of 98.75% and an average frame rate of 93. Compared to the initial model, the mean frame rate has been elevated by 7.23. The wildfire identification model designed in this paper improves the speed of identification while maintaining accuracy, thereby advancing the development and application of CNNs in the field of wildfire monitoring.