基于改进MobileNetV3的水稻病害识别模型
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金项目(32172780)


Rice Disease Identification Model Based on Improved MobileNetV3
Author:
Affiliation:

Fund Project:

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

    针对水稻病害识别方法准确度低、模型收敛速度缓慢的问题,本文提出了一种高性能的轻量级水稻病害识别模型,简称为CA(Coordinate attention)-MobileNetV3。通过微调的迁移学习策略完善了模型的训练,提升了模型收敛速度。首先创建10个种类的数据集,其中包含9种水稻病害和1种水稻健康叶片。其次使用CA模块,在通道注意力中嵌入空间坐标信息,提高模型的特征提取能力与泛化能力。最后,将改进后的MobileNetV3网络作为特征提取网络,并加入SVM多分类器,提高模型精度。实验结果表明,在本文构建的水稻病害数据集上,初始的MobileNetV3识别准确率仅为95.78%,F1值为95.36%;加入CA模块后识别准确率和F1值分别提高至96.73%和96.56%;再加入SVM多分类器,通过迁移学习后,改进模型的识别准确率和F1值分别达到97.12%和97.04%,参数量和耗时仅为2.99×106和0.91s,明显优于其他模型。本文提出的CA-MobileNetV3水稻病害识别模型能够有效识别水稻叶部病害,实现了轻量级、高性能、易部署的水稻病害分类识别算法。

    Abstract:

    For the problems of low accuracy of rice disease identification methods and slow convergence of models, a highperformance lightweight rice disease identification model was proposed, referred to as coordinate attention (CA)-MobileNetV3. The training of the model was optimized by fine-tuning the migration learning strategy, and the convergence speed of the model was improved. Firstly, a ten species dataset was created, containing nine rice diseases and healthy rice leaves. Secondly, the CA module was also used to embed spatial coordinate information in the channel attention to improve the feature extraction and generalization ability of the model. In addition, the improved MobileNetV3 network was used as the feature extraction network and the SVM multi-classifier was added to improve the model accuracy. The experimental results showed that on the rice disease dataset constructed, the initial MobileNetV3 recognition accuracy was only 95.78% and the F1 score was 95.36%, and then the recognition accuracy and F1 score were improved to 96.73% and 96.56%, respectively, after adding the CA module, and then the SVM multiclassifier was added, and the recognition accuracy and F1 scores reached 97.12% and 97.04%, respectively, the number of parameters and the time taken were only 2.99×106 and 0.91s, which were significantly better than that of other models. The experimental results showed that the CA-MobileNetV3 rice disease recognition model proposed can effectively recognize rice leaf diseases and achieve a lightweight, high-performance and easy-to-deploy rice disease classification and recognition algorithm.

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

崔金荣,魏文钊,赵敏.基于改进MobileNetV3的水稻病害识别模型[J].农业机械学报,2023,54(11):217-224,276. CUI Jinrong, WEI Wenzhao, ZHAO Min. Rice Disease Identification Model Based on Improved MobileNetV3[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(11):217-224,276.

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