Detection of Pine Wilt Disease in UAV Remote Sensing Images Based on SLMW-Net

The pine wood nematode is responsible for pine wilt disease, which poses a significant threat to forest ecosystems worldwide. If not quickly detected and removed, the disease spreads rapidly. Advancements in UAV and image detection technologies are crucial for disease monitoring, enabling efficient and automated identification of pine wilt disease. However, challenges persist in the detection of pine wilt disease, including complex UAV imagery backgrounds, difficulty extracting subtle features, and prediction frame bias. In this study, we develop a specialized UAV remote sensing pine forest ARen dataset and introduce a novel pine wilt disease detection model, SLMW-Net. Firstly, the Self-Learning Feature Extraction Module (SFEM) is proposed, combining a convolutional operation and a learnable normalization layer, which effectively solves the problem of difficult feature extraction from pine trees in complex backgrounds and reduces the interference of irrelevant regions. Secondly, the MicroFeature Attention Mechanism (MFAM) is designed to enhance the capture of tiny features of pine trees infected by initial nematode diseases by combining Grouped Attention and Gated Feed-Forward. Then, Weighted and Linearly Scaled IoU Loss (WLIoU Loss) is introduced, which combines weight adjustment and linear stretch truncation to improve the learning strategy, enhance the model performance and generalization ability. SLMW-Net is trained on the self-built ARen dataset and compared with seven existing methods. The experimental results show that SLMW-Net outperforms all other methods, achieving an mAP@0.5 of 86.7% and an mAP@0.5:0.95 of 40.1%. Compared to the backbone model, the mAP@0.5 increased from 83.9% to 86.7%. Therefore, the proposed SLMW-Net has demonstrated strong capabilities to address three major challenges related to pine wilt disease detection, helping to protect forest health and maintain ecological balance.