基于迁移学习的卷积神经网络植物叶片图像识别方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

中央高校基本科研业务费专项资金项目(2015ZCQ-GX-04)、国家自然科学基金项目(31670719)和北京市科技计划项目(Z161100000916012)


Plant Leaf Image Recognition Method Based on Transfer Learning with Convolutional Neural Networks
Author:
Affiliation:

Fund Project:

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

    为了提高植物叶片图像的识别准确率,考虑到植物叶片数据库属于小样本数据库,提出了一种基于迁移学习的卷积神经网络植物叶片图像识别方法。首先对植物叶片图像进行预处理,通过对原图的随机水平、垂直翻转、随机缩放操作,扩充植物叶片图像数据集,对扩充后的叶片图像数据集样本进行去均值操作,并以4∶1的比例划分为训练集和测试集;然后将训练好的模型(AlexNet、InceptionV3)在植物叶片图像数据集上进行迁移训练,保留预训练模型所有卷积层的参数,只替换最后一层全连接层,使其能够适应植物叶片图像的识别;最后将本文方法与支持向量机(SVM)方法、深度信念网络(DBN)方法、卷积神经网络(CNN)方法在ICL数据库进行对比实验。实验使用Tensorflow训练网络模型,实验结果由TensorBoard可视化得到的数据绘制而成。结果表明,利用AlexNet、InceptionV3预训练模型得到的测试集准确率分别为95.31%、95.40%,有效提高了识别准确率。

    Abstract:

    The identification of plant species plays an important role in the research and development of botany. In order to improve the recognition accuracy of plant leaf images, taking into account the plant leaf database as a small sample database, a convolutional neural network plant leaf image recognition method based on transfer learning was proposed. Firstly, the plant leaf image was preprocessed, and the plant leaf image data set was expanded by random horizontal, vertical flipping, and random scaling of the original image. The expanded plant leaf image data set samples were de-averaged and divided into 4∶1 proportions as training set and test set. Then, the pre-trained models (AlexNet and InceptionV3) on the large-scale datasets were loaded to the plant leaf image data set. The parameters of all convolution layers of the pre-training models were retained, but the last layer was completely replaced to adapt the task of plant leaf image recognition. Finally, the proposed methods were compared with the support vector machine (SVM) method, the deep belief nets (DBN) method and the convolutional neural network (CNN) method from training time and the accuracy of test set in the ICL database. Tensorflow was used to train the network model in the experiment, and the model was visualized by the Tensorboard. The results showed that the accuracy rate of test set obtained with AlexNet and InceptionV3 pre-training models were 95.31% and 95.40%, respectively, which were better than those with SVM method, DBN method and CNN method. The training time of AlexNet and InceptionV3 pre-training model were 26220s and 3984s, respectively. The InceptionV3 pre-training model was superior to the AlexNet pre-training model in terms of test set accuracy and training time.

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

郑一力,张露.基于迁移学习的卷积神经网络植物叶片图像识别方法[J].农业机械学报,2018,49(s1):354-359.

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