基于改进EfficientNetv2模型的多品种南药叶片分类方法
作者:
作者单位:

1.华南农业大学电子工程学院(人工智能学院),广州 510642;2.广东省农情信息监测工程技术研究中心,广州 510642

作者简介:

孙道宗,E-mail:sundaozong@scau.edu.cn

通讯作者:

王卫星, E-mail:weixing@scau.edu.cn

中图分类号:

S571.1

基金项目:

广东省现代农业关键技术模式集成与示范推广项目(粤财农【2021】37号-200011);国家自然科学基金项目(31671591;31971797);广州市科技计划项目(202002030245);广东省科技专项(2020020103);广东省现代农业产业技术体系创新团队建设专项(2021KJ108);广东省教育厅特色创新类项目(2019KTSCX013);2020年广东省科技创新战略专项(pdjh2020a0084);广东省大学生创新创业项目(S202010564150;202110564042)


Classification of leaves of multi-variety southern traditional Chinese medicine based on improved EfficientNetv2 model
Author:
Affiliation:

1.College of Electronic Engineering(College of Artificial Intelligence),South China Agricultural University,Guangzhou 510642,China;2.Guangdong Engineering Research Center for Monitoring Agricultural Information, Guangzhou 510642, China

Fund Project:

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

    为提高南药叶片的分类和分拣效率,本研究对 EfficientNetv2网络模型进行改进,引入迁移学习机制训练模型,选取自适应矩估计优化算法,通过多次测试进行超参数优化,确定学习率;采用MultiMarginLoss损失函数改善复杂背景信息对识别效果的影响。应用改进后的EfficientNetv2模型与其他轻量级模型对实地采集的复杂背景下的8种南药叶片进行分类效果对比试验,试验结果显示,改进模型对复杂背景下的南药叶片图像样本识别准确率为99.12%,相较于初始模型EfficientNetv2-S,准确率提高1.17%,并且参数量和模型大小均下降约85%,平均训练时间下降47.62%。与DenseNet121、ShuffleNet和RegNet等模型相比,改进模型在模型存储空间大小、准确率和训练时间3个指标上有明显优势。研究结果表明,在多品种南药叶片分类任务中,改进模型取得优良表现,模型的轻量化程度和性能得到进一步的提升。

    Abstract:

    For the application of deep learning in the field of Chinese herbal medicine, especially in the field of southern Chinese medicine, the complex background will reduce the accuracy of recognition.If a complex network structure is used, high computing power is required to support training and detection, but the actual embedded or mobile devices are difficult to meet, which affects the effect of on-site detection.This article proposed to improve the EfficientNetv2 network model to classify and identify 8 kinds of southern Chinese medicine leaves in the complex background collected in the field.The network structure was redesigned.The scope of the Fused-MBConv and MBConv architectures was adjusted.Some 3×3 convolutional kernels were replaced with 5×5 convolutional kernels to increase the perceptual field size, reduce the number of convolutional layers of the network, and to further reduce the network complexity.The transfer learning was introduced to train the model.The adaptive moment estimation optimization algorithm was used to optimize the hyperparameters with multiple tests to determine the learning rate.MultiMarginLoss was selected as the loss function to solve the problem of complex background information affecting the accuracy of recognition.The diversity of the data set was increased and the problem of model over-fitting was avoided by adopting data augmentation methods including affine transformation and Gaussian blur and other methods to the experimental data set to improve the stability of the model training process.The results showed that the model improved achieved 99.12% accuracy in recognizing the image of southern Chinese medicine leaves with complex backgrounds, 1.17% more accurate than the baseline model EfficientNetv2-S.The size of parameters was reduced by 85% approximately.The average training time was reduced by 47.62%.The improved model had significant advantages in model storage space, accuracy, and training time, comparing with lightweight models including DenseNet121, ShuffleNet, and RegNet.It is indicated that the model proposed performs well in classifying leaves of multi-variety southern Chinese medicine.The lightweight degree and performance of the model are further improved.

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

孙道宗,刘锦源,丁郑,刘欢,彭家骏,谢家兴,王卫星.基于改进EfficientNetv2模型的多品种南药叶片分类方法[J].华中农业大学学报,2023,42(1):258-267

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