1College of Forestry, Nanjing Forestry University, Nanjing 210037, China.
2Co-Innovation Center for Sustainable Forestry in Southern China, Nanjing Forestry University, Nanjing 210037, China
| Received 05 Sep 2023 |
Accepted 23 Nov 2023 |
Published 15 Dec 2023 |
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.
