针对茶树病虫害检测中的遮挡问题,提出改进的YOLOv8模型,提升小目标与遮挡区域的识别准确率。
DAONet-YOLOv8: An Occlusion-Aware Dual-Attention Network for Tea Leaf Pest and Disease Detection
- 融合局部卷积与全局注意力,聚焦病斑区域并抑制背景噪声
- 设计遮挡感知检测头,利用可见部分推断被遮挡病斑特征
- 引入动态多核卷积模块,更好捕捉不规则病斑边界,参数减少16.7%
由于复杂背景、光照变化及密集枝叶频繁遮挡,真实茶园中茶树病虫害的精准检测仍具挑战,现有检测器常出现漏检和误报。为此,本文提出DAONet-YOLOv8,通过三项改进:(1) 双注意力融合模块(DAFM)结合卷积局部特征提取与自注意力全局上下文建模,聚焦细微病斑并抑制背景噪声;(2) 遮挡感知检测头(Detect-OAHead),学习可见与被遮挡部分关系以补偿缺失病斑特征;(3) 采用动态合成卷积的C2f-DSConv模块,通过多核形状适应不规则病斑边界。在包含六类病虫害的真实茶园数据集上,该模型达到92.97%精度、92.80%召回率、97.10% mAP@50、76.90% mAP@50:95,较YOLOv8n分别提升2.34、4.68、1.40、1.80个百分点,同时参数量减少16.7%。对比实验进一步验证其优于主流检测模型。
原文摘要 · Abstract (English)
Accurate detection of tea leaf pests and diseases in real plantations remains challenging due to complex backgrounds, variable illumination, and frequent occlusions among dense branches and leaves. Existing detectors often suffer from missed detections and false positives in such scenarios. To address these issues, we propose DAONet-YOLOv8, an enhanced YOLOv8 variant with three key improvements: (1) a Dual-Attention Fusion Module (DAFM) that combines convolutional local feature extraction with self-attention based global context modeling to focus on subtle lesion regions while suppressing background noise; (2) an occlusion-aware detection head (Detect-OAHead) that learns the relationship between visible and occluded parts to compensate for missing lesion features; and (3) a C2f-DSConv module employing dynamic synthesis convolutions with multiple kernel shapes to better capture irregular lesion boundaries. Experiments on our real-world tea plantation dataset containing six pest and disease categories demonstrate that DAONet-YOLOv8 achieves 92.97% precision, 92.80% recall, 97.10% mAP@50 and 76.90% mAP@50:95, outperforming the YOLOv8n baseline by 2.34, 4.68, 1.40 and 1.80 percentage points respectively, while reducing parameters by 16.7%. Comparative experiments further confirm that DAONet-YOLOv8 achieves superior performance over mainstream detection models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。