|
|
Improved YOLOv5s UAV aerial forest fire smoke detection |
|
|
Abstract To solve the problems of low detection accuracy and high calculation cost of traditional forest fire smoke detection, a forest fire smoke detection algorithm based on improved YOLOv5s is proposed. A fire and smoke dataset containing 16573 images is constructed to solve the problem of insufficient existing data sets and improve the generalization ability of the training model. A lightweight GC-C3 module is designed to replace the original C3 module and reduce the number of model parameters and calculation. The weighted bidirectional feature pyramid network is integrated into the Neck structure to enhance the detection ability of the network for small and medium targets. The network space pyramid pool structure is modified, SPPF is replaced by SimSPPF structure, and the computing efficiency and detection accuracy of the network are improved. The bounding box regression loss function CIOU is replaced by Focal-EIOU to accelerate the convergence of the model and solve the problem of mismatch between positive and negative samples. The experimental results show that the average detection accuracy of the improved network is increased by 2.3%, the number of model parameters is decreased by 46.7%, and the calculation amount of the model is decreased by 47.5%.
|
Received: 04 February 2024
Published: 26 September 2024
|
|
|
|
|
|
|