基于改进YOLO v5算法的草莓缺素诊断方法

扫码查看

原文链接

万方数据
维普

中文摘要：为解决实际生产中草莓因缺素而导致经济损失的问题,提出一种基于YOLO v5算法的草莓叶片无损缺素检测方法,可针对4种常见的缺素(缺氮、缺磷、缺钾、缺钙)草莓叶片及正常草莓叶片进行识别.由于草莓的种植环境较为复杂,因此对YOLO v5算法进行改进,包括在骨干网络中添加CBAM注意力机制、使用Focal-EIoU损失函数替换默认的CIoU损失函数,以及引入Soft-NMS非极大值抑制算法并加入P2检测头,这些改进着重增加了算法在复杂背景下针对重叠目标及小目标的检测能力.经过改进所建立的YOLO v5-CFPS模型的准确率(P)、召回率(R)、平均精度均值(mAP值)分别为97.05％、95.71％、97.03％,相较于原始YOLO v5模型分别提升了 6.47、6.01、7.73百分点,并通过NCNN框架将模型移植至安卓平台,验证了其实际应用的可行性.说明YOLO v5-CFPS模型对于草莓叶片缺素检测具有检测精度高、速度快、可靠性强等优点,为草莓缺素检测提供了一种更优的解决方案,有助于实现及时施肥追肥,并减少因不科学用肥所导致的经济损失和资源浪费.

作者：

王克林、王成义、郭冰、李志勇、李军台、丁筱玲

展开 >

作者单位：

山东农业大学机械与电子工程学院/山东省园艺机械与装备重点实验室,山东泰安 271018

山东农业大学信息科学与工程学院,山东泰安 271018

关键词：

缺素诊断草莓叶片机器视觉改进YOLO v5算法注意力机制

出版年：

2024

DOI：

10.15889/j.issn.1002-1302.2024.20.011