BD-YOLOv8: a Lightweight Method for Real-Time Citrus Disease Detection in Precision Agriculture
摘要
Real-time monitoring of fruit growth in natural environments is an essential component of precision agriculture, encompassing disease control and identification. Existing detection algorithms face challenges such as a large number of parameters, thereby hindering their deployment on embedded devices. To address the internal complexity of the model, a lightweight detection algorithm, BD-YOLOv8, has been developed. This algorithm utilizes three collaborative modules to achieve precise classification and localization of diseases. Firstly, a novel image edge feature extraction module (IEFEM) was proposed. This enables the model to better extract image content by combining Sobel convolution with spatial information convolution. Moreover, an innovative detail-capture block (DC_Block) was introduced to improve the model’s perceptual ability and detection accuracy. Ultimately, a BN-separated parameter-sharing detection head (BSPSDH) was designed. This achieves a lightweight construction without loss of accuracy by sharing parameters between convolutional layers. The results indicate that BD-YOLOv8 achieves an mAP50 of 98.2% and an FPS of 149.6. Additionally, there were respective reductions of 42.4% in parameters, 27.2% in GFLOPs, and 40.3% in model size.