河南农业科学 ›› 2021, Vol. 50 ›› Issue (4): 174-180.DOI: 10.15933/j.cnki.1004-3268.2021.04.023

所属专题: 作物图像采集与识别专题

• 农业信息与工程·农产品加工 • 上一篇    

基于轻量级卷积神经网络和迁移学习的小麦叶部病害图像识别

冯晓1,2,李丹丹1,王文君3,郑国清1,2,刘海礁1,2,孙永胜1,梁山1,杨莹1,臧贺藏1,2,张辉1,2   

  1. (1.河南省农业科学院 农业经济与信息研究所,河南 郑州 450002;2.河南省智慧农业工程技术研究中心,河南 郑州 450002;3.河南省林业科学研究院,河南 郑州 450008)
  • 收稿日期:2020-11-02 出版日期:2021-04-15 发布日期:2021-04-15
  • 通讯作者: 张辉(1975-),男,河南光山人,副研究员,本科,主要从事农业信息系统研发、大数据分析挖掘技术研究。E-mail:nkyzhanghui@163.com
  • 作者简介:冯晓(1978-),女,河南郑州人,副研究员,硕士,主要从事农业大数据分析及图像处理技术研究。E-mail:308564967@qq.com
  • 基金资助:
    河南省科技攻关计划项目(162102210377,182102110047,212102110253)

Image Recognition of Wheat Leaf Diseases Based on Lightweight Convolutional Neural Network and Transfer Learning

FENG Xiao1,2,LI Dandan1,WANG Wenjun3,ZHENG Guoqing1,2,LIU Haijiao1,2,SUN Yongsheng1,LIANG Shan1,YANG Ying1,ZANG Hecang1,2, ZHANG Hui1,2   

  1. (1.Institute of Agricultural Economics and Information,Henan Academy of Agricultural Sciences,Zhengzhou 450002,China;2.Henan Engineering and Technology Research Center for Intelligent Agriculture,Zhengzhou 450002,China;3.Henan Academy of Forestry,Zhengzhou 450008,China)
  • Received:2020-11-02 Published:2021-04-15 Online:2021-04-15

摘要: 为实现基于移动端的小麦叶部病害图像便捷识别,基于轻量级卷积神经网络(Convolutional neural network,CNN)和迁移学习建立小麦叶部病害图像识别模型。首先,建立由小麦白粉病、条锈病和叶锈病3种小麦叶部病害图像组成的样本集,每幅图像大小为224像素×224像素;然后,采用深度学习框架Tensorflow 2.0,基于MobileNetV2构建小麦叶部病害图像识别模型,使用ImageNet数据集上训练好的参数作为模型初始参数;最后,分析迁移学习方法、样本量、全局平均池化(Global average pooling,GAP)前添加Dropout层、初始学习率大小对模型性能的影响。结果表明,采用将模型所有层设置为可训练的迁移学习方式、选择适合的数据增强方法增加样本量、在GAP前添加Dropout层、设置0.000 01的初始学习率,对3种小麦病害图像的平均识别准确率高达99.96%。可见,基于MobileNetV2和迁移学习可构建识别准确率高、泛化能力强、适合移动端应用的小麦叶部病害图像识别模型。

关键词: 小麦;叶部病害, 卷积神经网络;迁移学习;图像识别;MobileNetV2;计算机视觉

Abstract: In order to realize the convenient recognition of wheat leaf disease image on mobile terminal,image recognition model of wheat leaf disease was established based on lightweight convolutional neural network (CNN) and transfer learning.Firstly,a sample data set of 224 pixel×224 pixel was established,which was composed of images of wheat powdery mildew,stripe rust and leaf rust. Secondly,the deep learning framework Tensorflow 2.0 was used to build the wheat leaf disease image recognition model based on MobileNetV2, and the trained parameters on ImageNet data set were used as the initial parameters of the model. Finally,the effects of transfer learning method,sample size,adding Dropout layer before global average pooling and initial learning rate setting on the model performance were analyzed.The results showed that the average recognition accuracy of the images of three wheat diseases was as high as 99.96% by setting all layers of the model as trainable,selecting suitable data enhancement method to increase the sample size, adding Dropout layer before global average pooling and setting the initial learning rate of 0.000 01.Based on MobileNetV2 and transfer learning, a wheat leaf disease image recognition model with high recognition accuracy and strong generalization ability could be constructed for mobile terminal.

Key words: Wheat, Leaf diseases;Convolutional neural network;Transfer learning;Image recognition;MobileNetV2;Computer vision

中图分类号: