基于YOLO v5s的作物叶片病害检测模型轻量化方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家重点研发计划项目(2022YFD1900801)


Lightweight Method for Crop Leaf Disease Detection Model Based on YOLO v5s
Author:
Affiliation:

Fund Project:

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

    为在保证识别性能前提下,对叶片病害检测模型进行有效轻量化,基于主干替换、模型剪枝以及知识蒸馏技术构建了一种模型轻量化方法,对以YOLO v5s为基础的叶片黄化曲叶病检测模型开展轻量化试验。首先,通过常见的性能优异的轻量级主干特征提取神经网络结构(Lightweight convolutional neural networks,LCNN)替换YOLO v5s主干对模型主体进行缩减;然后利用模型稀疏化训练和批归一化层(Batch normalization layer)的缩放因子分布状况,筛选并删减不重要的通道;最后,通过微调重新训练以及知识蒸馏,将模型精度调整到接近剪枝前的水平。试验结果表明,经轻量化处理的模型精确率、召回率和平均精度分别为91.3%、87.4%和92.7%,模型内存占用量为1.4 MB,台式机检测帧率81.0f/s,移动端检测帧率1.2f/s,相比原始YOLO v5s叶片病害检测模型,精确率、召回率和平均精度下降3.7、4.6、2.7个百分点,内存占用量仅为处理前的10%,台式机和移动端检测的帧率分别提升近27%和33%。本文所提出的方法在保持模型性能的前提下对模型有效轻量化,为移动端叶片病害检测部署提供了理论基础。

    Abstract:

    In order to effectively lightweight the leaf disease detection model under the premise of ensuring the recognition performance, a model lightweight method was constructed based on trunk replacement, model pruning and knowledge distillation technology, and a lightweight test was carried out on the leaf yellow leaf curl disease detection model based on YOLO v5s. Firstly, the main body of the model was reduced by replacing the YOLO v5s trunk with the common lightweight convolutional neural networks (LCNN) with excellent performance. Then, the unimportant channels were screened and deleted by using the sparse training of the model and the distribution of the scaling factors in the batch normalization layer. Finally, by fine-tuning retraining and knowledge distillation, the model accuracy was adjusted to a level close to that before pruning. The experimental results showed that the accuracy, recall and mean average accuracy of the lightweight model were 91.3%, 87.4% and 92.7%, respectively. The memory consumption of the model was 1.4MB, and the detection frame rate of the desktop was 81.0f/s. The detection frame rate of the mobile terminal was 1.2f/s. Compared with the original YOLO v5s leaf disease detection model, the accuracy, recall and average accuracy were reduced by 3.7 percentage points, 4.6 percentage points and 2.7 percentage points, and the memory consumption was only 10% of that before processing. The frame rate of the desktop and mobile terminal detection was increased by nearly 27% and 33%, respectively. The proposed method can effectively reduce the weight of the model under the premise of keeping the performance, which provided a theoretical basis for the deployment of mobile leaf disease detection.

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

杨佳昊,左昊轩,黄祺成,孙泉,李思恩,李莉.基于YOLO v5s的作物叶片病害检测模型轻量化方法[J].农业机械学报,2023,54(s1):222-229. YANG Jiahao, ZUO Haoxuan, HUANG Qicheng, SUN Quan, LI Sien, LI Li. Lightweight Method for Crop Leaf Disease Detection Model Based on YOLO v5s[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(s1):222-229.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-06-28
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-12-10
  • 出版日期: