基于改进YOLO v5的水稻主要害虫识别方法

作者:吴子炜; 夏芳*; 陆林峰; 张盛军; 周素茵
来源:江苏农业科学, 2023, 51(21): 218-224.
DOI:10.15889/j.issn.1002-1302.2023.21.033

摘要

准确识别害虫是水稻田间害虫防控的基础,针对现有水稻害虫识别方法精度不高、计算量与参数量较大等问题,提出一种轻量模块以及相应改进YOLO v5模型的方法。为了在满足精度的前提下减小模型参数量,构建的轻量模块使用多个深度可分离卷积减小计算消耗,并通过通道洗牌将特征通道重组提高学习能力,同时将该模块引入YOLO v5模型中进行轻量化改进。为提高模型的泛用性,除选取公开数据集IP102中的水稻害虫图片外,通过二化螟性诱设备采集二化螟图像丰富研究数据。结果表明,所提出的轻量化模块能够有效减轻模型的体积、优化训练并提升模型的识别精度,改进后的模型参数量仅为原模型的一半,对14类害虫的平均识别精度相较原模型提高3.2百分点,拥有一定的实际应用能力。所提出的轻量模块及轻量化改进模型能够实现对水稻害虫高精度及高效率的识别,可为水稻病虫害防控数字化、智能化发展提供技术支持。

  • 出版日期2023
  • 单位浙江农林大学; 浙江托普云农科技股份有限公司; 经济管理学院; 浙江省农业

全文