Forest Fire Detection Algorithm Based on Improved YOLOv11n

To address issues in traditional forest fire detection models, such as large parameter sizes, slow detection speed, and unsuitability for resource-constrained devices, this paper proposes a forest fire detection method, FEDS-YOLOv11n, based on an improved YOLOv11n model. First, the C3k2 module was redesigned using the FasterBlock module, replacing C3k2 with C3k2-Faster in both the Backbone network and Neck section to achieve a lightweight model design. Second, an EMA attention mechanism was introduced into the C3k2-Faster module in the Backbone, replacing C3k2-Faster with C3k2-Faster-EMA to compensate for the accuracy loss in small-object detection caused by the lightweight design. Third, the original upsampling module in the Neck was replaced with the lightweight dynamic upsampling operator DySample. Finally, the detection head was improved using the SEAM attention module, replacing the original Detect head with SEAMHead, which enables better handling of occluded objects. The experimental results show that compared to YOLOv11n, the proposed FEDS-YOLOv11n achieves improvements of 0.9% in precision (P), 1.9% in recall (R), 2.1% in mean precision at IoU 0.5 ([email protected]), and 2.3% in mean precision at IoU 0.5-0.95 ([email protected]). Additionally, the number of parameters is reduced by 21.32%, GFLOPs are reduced by 26.98%, and FPS increases from 48.2 to 71.8. The FEDS-YOLOv11n model ensures high accuracy while maintaining lower computational complexity and faster inference speed, making it suitable for real-time forest fire detection applications.