FGYOLO: An Integrated Feature Enhancement Lightweight Unmanned Aerial Vehicle Forest Fire Detection Framework Based on YOLOv8n

• Published in: Forests Vol. 15; no. 10; p. 1823
• Main Authors: Zheng, Yangyang, Tao, Fazhan, Gao, Zhengyang, Li, Jingyan
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 01.10.2024
• Subjects: Accuracy; Algorithms; Datasets; Deep learning; Drone aircraft; Forest & brush fires; Forest fire detection; Forest fires; Modules; Parameters; Satellites; Smoke; Target detection; Target recognition; Unmanned aerial vehicles
• ISSN: 1999-4907; 1999-4907
• DOI: 10.3390/f15101823

To address the challenges of complex backgrounds and small, easily confused fire and smoke targets in Unmanned Aerial Vehicle (UAV)-based forest fire detection, we propose an improved forest smoke and fire detection algorithm based on YOLOv8. Considering the limited computational resources of UAVs and the lightweight property of YOLOv8n, the original model of YOLOv8n is improved, the Bottleneck module is reconstructed using Group Shuffle Convolution (GSConv), and the residual structure is improved, thereby enhancing the model’s detection capability while reducing network parameters. The GBFPN module is proposed to optimize the neck layer network structure and fusion method, enabling the more effective extraction and fusion of pyrotechnic features. Recognizing the difficulty in capturing the prominent characteristics of fire and smoke in a complex, tree-heavy environment, we implemented the BiFormer attention mechanism to boost the model’s ability to acquire multi-scale properties while retaining fine-grained features. Additionally, the Inner-MPDIoU loss function is implemented to replace the original CIoU loss function, thereby improving the model’s capacity for detecting small targets. The experimental results of the customized G-Fire dataset reveal that FGYOLO achieves a 3.3% improvement in mean Average Precision (mAP), reaching 98.8%, while reducing the number of parameters by 26.4% compared to the original YOLOv8n.