中国农业机械化科学研究院集团有限公司 主管

北京卓众出版有限公司 主办

基于改进YOLOv8的葡萄叶片病害检测模型

Grape leaf disease detection model based on improved YOLOv8

  • 摘要: 葡萄叶片病害的精准快速检测对保障葡萄产业健康发展至关重要。针对复杂田间环境下病害目标小、形态多变、背景干扰强等检测难题,提出一种基于改进YOLOv8的目标检测模型。首先,引入自适应空间相关性金字塔注意力机制(ASCPA),通过多尺度特征融合与自适应权重分配,显著增强模型对小尺寸病斑的细节捕捉与特征区分能力。其次,采用Wise-IOU损失函数替代传统边界框回归损失,通过动态聚焦机制优化不同质量样本的梯度更新,有效提升病斑的定位精度,特别是在重叠和边缘模糊区域。最后,使用一种带热身的余弦退火调度器调整学习率与批处理协同优化策略,在引入复杂模块增加模型容量的同时,有效防止过拟合并大幅加速训练收敛。在自建的葡萄叶片病害数据集上进行试验,结果表明,改进YOLOv8的精确率(Precision)、召回率(Recall)和全类别平均精度均值(mAP50)分别达到0.99228、0.99345和0.99384,较基线模型提升显著。同时,训练效率提高约38%,实现了精度与效率的同步优化,为田间病害智能检测提供了有效的解决方案。

     

    Abstract: Accurate and rapid detection of grape leaf diseases is crucial for ensuring grape industry healthy development.Aiming at detection challenges in complex field environments, such as small disease targets, variable morphologies, and strong background interference, an target detection model based on improved YOLOv8 has been proposed.Firstly, an adaptive spatial correlation pyramid attention(ASCPA)mechanism has been introduced, which significantly enhanced model's ability to capture details and distinguish features of small-sized lesions through multi-scale feature fusion and adaptive weight allocation.Secondly, Wise-IOU loss function has been employed to replace traditional bounding box regression loss.By dynamically focusing mechanism on gradient updates for different qualities' samples, this approach has effectively improved lesions' localization accuracy, especially in overlapping and edge-blurred areas.Finally, a warm-up cosine annealing scheduler has been used to adjust learning rate and batch collaborative optimization strategy, effectively preventing overfitting and significantly accelerating training convergence, while introducing complex modules to increase model capacity.Experimental were conducted on a self-built grape leaf disease dataset, and results have shown that improved model achieved precision, recall, and mean average precision(mAP50)of 0.99228, 0.99345, and 0.99384, respectively, representing significant improvements over baseline model.Furthermore, training efficiency has been increased by approximately 38%, which achieved simultaneous optimization of accuracy and efficiency, providing an effective solution for intelligent disease detection in field.

     

/

返回文章
返回