Exploring the Close-Range Detection of UAV-Based Images on Pine Wilt Disease by an Improved Deep Learning Method

Abstract

Pine wilt disease (PWD) is a significantly destructive forest disease. To control the spread of PWD, an urgent need exists for a real-time and efficient method to detect infected trees. However, existing object detection models have often faced challenges in balancing lightweight design and accuracy, particularly in complex mixed forests. To address this, an improvement was made to the YOLOv5s (You Only Look Once version 5s) algorithm, resulting in a real-time and efficient model named PWD-YOLO. First, a lightweight backbone was constructed, composed of multiple connected RepVGG Blocks, significantly enhancing the model’s inference speed. Second, a C2fCA module was designed to incorporate rich gradient information flow and concentrate on key features, thereby preserving more detailed characteristics of PWD-infected trees. In addition, the GSConv network was utilized instead of conventional convolutions to reduce network complexity. Last, the Bidirectional Feature Pyramid Network strategy was used to enhance the propagation and sharing of multiscale features. The results demonstrate that on a self-built dataset, PWD-YOLO surpasses existing object detection models with respective measurements of model size (2.7 MB), computational complexity (3.5 GFLOPs), parameter volume (1.09 MB), and speed (98.0 frames/s). The Precision, Recall, and F1-score on the test set are 92.5%, 95.3%, and 93.9%, respectively, which confirms the effectiveness of the proposed method. It provides reliable technical support for daily monitoring and clearing of infected trees by forestry management departments.