基于改进YOLOv3的水稻叶部病害检测
DOI:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

天津市科技支撑计划项目(19YFZCSN00360)


Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为了解决水稻小病斑检测不准确的问题,提出一种基于改进YOLOv3的水稻叶部病害检测方法Rice–YOLOv3。首先,采用K–means++聚类算法,计算新的锚框尺寸,使锚框尺寸与数据集相匹配;其次,采用激活函数Mish替换YOLOv3主干网络中的Leaky Relu激活函数,利用该激活函数的平滑特性,提升网络的检测准确率,同时将CSPNet与DarkNet53中的残差模块相结合,在避免出现梯度信息重复的同时,增加神经网络的学习能力,提升检测精度和速率;最后,在FPN层分别引入注意力机制ECA和CBAM模块,解决特征层堆叠处的特征提取问题,提高对小病斑的检测能力。在训练过程中,采用COCO数据集预训练网络模型,得到预训练权重,改善训练效果。结果表明:在测试集下,Rice–YOLOv3检测水稻叶部3种病害的平均精度均值(mAP)达92.94%,其中,稻瘟病、褐斑病、白叶枯病的mAP值分别达93.34%、89.68%、95.80%,相较于YOLOv3,Rice–YOLOv3检测的mAP提高了6.05个百分点,速率提升了2.8帧/s,对稻瘟病和褐斑病的小病斑的检测能力明显增强,可以检测出原始网络模型漏检的小病斑;与Faster–RCNN、YOLOv5等模型对比,Rice–YOLOv3提高了对相似病害和微小病害的识别能力,并在原始的基础上提高了检测速率。

    Abstract:

    In order to solve the problem of inaccurate detection of small spots in rice, a rice leaf disease detection method Rice-YOLOv3 based on the improved YOLOv3 was proposed in this study. First, the K-means++ clustering algorithm was used to compute the new anchor frame size for data matching. Second, the activation function Mish was used to replace the Leaky Relu activation function in the YOLOv3 backbone network with a goal to improve the detection accuracy of the network by use of the smoothing property. And, the CSPNet was combined with the residual module in DarkNet53 to avoid the repetition of the gradient information and increase the learning ability of the neural network to improve the detection accuracy and speed. Finally, the attention mechanism ECA and CBAM modules were introduced at the FPN layer to solve the feature extraction problem at the feature layer stacking and improve the detection ability of small spots. In the training process, the COCO dataset was used to pre-train the network model to get the pre-training weights and improve the training effect. The results showed that the mean average precision mean(mAP) of Rice-YOLOv3 for in the rice leaf disease amounted to 92.94%, of which the mAP values of rice blast, brown spot and leaf blight reached 93.34%, 89.68%, 95.80%, respectively. Compared to the YOLOv3, the mAP of Rice-YOLOv3 detection increased by 6.05 percentage points and the speed was improved by 2.8 frames/s, and the detection ability of small spots of rice blast and brown spot was significantly enhanced including those small spots missed by the original network model. Comparing with the models of Faster-RCNN, YOLOv5, etc., the Rice-YOLOv3 improved the ability of recognizing the similar and tiny diseases as well as the detection speed.

    参考文献
    相似文献
    引证文献
引用本文

赵辉,李建成,王红君,岳有军.基于改进YOLOv3的水稻叶部病害检测[J].湖南农业大学学报:自然科学版,2024,50(1):.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-03-22
  • 出版日期:
文章二维码