基于迁移学习的卷积神经网络玉米病害图像识别
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家重点研发计划项目(2017YFC0403203)和陕西省水利科技计划项目(2014slkj-18)


Recognition of Corn Leaf Spot and Rust Based on Transfer Learning with Convolutional Neural Network
Author:
Affiliation:

Fund Project:

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

    为实现小数据样本复杂田间背景下的玉米病害图像识别,提出了一种基于迁移学习的卷积神经网络玉米病害图像识别模型。在VGG-16模型的基础上,设计了全新的全连接层模块,并将VGG-16模型在ImageNet图像数据集训练好的卷积层迁移到本模型中。将收集到的玉米病害图像数据集按3∶1的比例分为训练集与测试集。为扩充图像数据,对训练集原图进行了旋转、翻转等操作。基于扩充前后的训练集,对只训练模型的全连接层和训练模型的全部层(卷积层+全连接层)两种迁移学习方式进行了试验,结果表明,数据扩充和训练模型的全部层能够提高模型的识别能力。在训练模型全部层和训练集数据扩充的条件下,对玉米健康叶、大斑病叶、锈病叶图像的平均识别准确率为95.33%。与全新学习相比,迁移学习能够明显提高模型的收敛速度与识别能力。将训练好的模型用Python开发为图形用户界面,可实现田间复杂背景下玉米大斑病与锈病图像的智能识别。

    Abstract:

    In order to realize the identification of corn disease images in complex field background for small data samples, a corneal disease image recognition model based on transfer learning was proposed. Based on the VGG-16 model, a new fully connected layer module was designed. The VGG-16 model was migrated to the model in the trained convolution layer of the ImageNet image data set. The collected corn disease image data set was divided into a training set and a test set according to a ratio of 3∶1. In order to expand the data set of the image, the original set of the training set was rotated, flipped, and the like. Based on the training set before and after the expansion, the two layers of the training model, the full connection layer and the training model, all the layers (convolution layer + full connection layer) were tested. The results showed that all the layers of the data expansion and training model can improve the recognition ability of the model. Under the condition of all the layers of the training model and the expansion of the training set data, the average recognition accuracy of the image of corn healthy leaves, large spot disease leaves and rust leaves was 95.33%. Compared with the new learning, transfer learning can significantly improve the convergence speed and recognition ability of the model. Finally, the trained model was developed into a visual user interface, which can realize the intelligent recognition of corn leaf spot and rust images in the complex background of the field.

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

许景辉,邵明烨,王一琛,韩文霆.基于迁移学习的卷积神经网络玉米病害图像识别[J].农业机械学报,2020,51(2):230-236,253. XU Jinghui, SHAO Mingye, WANG Yichen, HAN Wenting. Recognition of Corn Leaf Spot and Rust Based on Transfer Learning with Convolutional Neural Network[J]. Transactions of the Chinese Society for Agricultural Machinery,2020,51(2):230-236,253.

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