高级检索+

基于改进YOLOv8的葡萄叶片病害检测与识别

Grape leaf disease detection and identification based on improved YOLOv8

  • 摘要: 为了提高葡萄叶片病害检测的效率与准确性,克服传统人工检测方法效率低、主观性强的局限,该研究提出一种基于改进YOLOv8的葡萄叶片病害检测方法,通过引入CBAM注意力机制优化特征金字塔结构,增强对不同尺度病斑的特征捕获能力;同时采用ShuffleNet V2轻量级网络替代原有CBS模块,在保持检测性能的同时显著降低模型参数量和计算复杂度。试验结果表明,改进模型在包含3种主要病害(黑腐病、褐斑病、轮斑病)的5000张叶片图像数据集上实现了94.6%的平均精度均值(mAP@0.5),较YOLOv8提升2.7个百分点,推理速度52.06 FPS,满足田间实时检测需求。与YOLOv5、Faster R-CNN、YOLOv8等模型对比,改进模型在检测精度和效率方面均表现出优势,为葡萄叶片病害智能识别提供了一定支持,该模型可以作为葡萄叶片病害检测的一种有效方法。

     

    Abstract: In China, grapes are not only a common fresh fruit but also the primary ingredient for winemaking, with a continuously growing market demand and cultivation area. However, grapes are susceptible to diseases during their growth period, which not only affects the quality and yield of the grapes but also leads to economic losses for farmers.Once grapevines are affected by diseases, their physiological and morphological characteristics change, with these changes often being prominently reflected in the leaves.Therefore, applying computer vision and deep learning technologies for disease detection and identification in grape leaves can achieve rapid detection and identification of grape leaf diseases, reducing the risk of cultivation.To enhance the precision and processing speed of grape leaf spot recognition, we propose an innovative detection method based on the optimization of the YOLOv8 algorithm.YOLOv8 can select network models of different depths and widths according to task requirements for object detection, image classification, instance segmentation, and keypoint detection tasks.The neck network of YOLOv8 has not changed much, using a PAN+FPN structure to enhance the network's ability to fuse features. The head part is used to convert the output feature maps from the neck part into detection results. The head part of YOLOv8 draws on the decoupled head design from YOLOX and YOLOv6, eliminating the objectness branch, and performs bounding box regression and object classification through two parts: 4×reg_max and num_class.Firstly, we need to preprocess the images of grape leaf diseases. The potential uneven distribution of disease image samples may lead to model overfitting. To mitigate this, we employ data augmentation techniques, including image rotation and mirror transformation, to increase sample diversity. These measures collectively aim to improve the model's generalization capability and accuracy in disease detection.Secondly, this study aims to enhance the precision and processing speed of grape leaf spot recognition; therefore, the proposed method involves integrating the CBAM (Convolutional Block Attention Module) attention mechanism into the backbone to enhance the refined processing of feature maps.The Convolutional Block Attention Module (CBAM) is a concise and efficient attention mechanism designed as a module within feedforward convolutional networks, capable of effectively integrating spatial and channel information, thereby enhancing the network's recognition capabilities.Since downsampling during the feature extraction process may lead to information loss when dealing with spots of varying sizes, this approach can enhance the model's ability to recognize spots of different scales.Finally, we have adopted ShuffleNet V2, a lightweight convolutional neural network, as the main framework, replacing the traditional CBS convolutional module, to reduce the model's parameters and computational requirements.ShuffleNet V2 places special emphasis on the balanced distribution of parameters and computational load across different parts of the network, thereby optimizing resource utilization efficiency and reducing the model's energy consumption. The design of ShuffleNet V2 adheres to several core principles to ensure efficient computational performance: reducing memory access by simplifying operations and optimizing data paths, thereby lowering energy consumption and increasing speed; simplifying the network structure and avoiding the use of computationally expensive operations.This approach ensures detection speed while maintaining the model's high efficiency and accuracy, demonstrating a significant improvement in the detection of grape leaf spots.Experiments have demonstrated that the proposed method ensures detection speed while maintaining the model's high efficiency and accuracy, achieving a significant improvement in the detection performance of grape leaf spots.We trained the model using a dataset of images depicting three types of grape leaf diseases, and compared experiments with the YOLOv5, YOLOv8, and the improved YOLOv8 models, as well as other models. The results showed that the accuracy of the improved YOLOv8 model reached 94.6%, which is an enhancement compared to the accuracy of other models. This model can serve as an effective approach for the detection of grape leaf diseases.

     

/

返回文章
返回