计算机与现代化 ›› 2020, Vol. 0 ›› Issue (11): 39-46.

• 人工智能 • 上一篇    下一篇

基于生成对抗网络的农作物叶片病害识别

  

  1. (东华理工大学信息工程学院,江西南昌330013)
  • 出版日期:2020-12-03 发布日期:2020-12-03
  • 作者简介:熊方康(1995—),男,江西东乡人,硕士研究生,研究方向:图像处理,深度学习,E-mail: fkangxiong@yeah.net; 陆玲(1962—),女,湖南衡阳人,教授,硕士生导师,研究方向:计算机图形学,数字图像处理,计算机可视化; 曹廷荣(1988—),男,硕士研究生,研究方向:图像处理,机器学习; 彭丽君(1995—),女,硕士研究生,研究方向:计算机图形学。
  • 基金资助:
    国家自然科学基金资助项目(61761003); 东华理工大学研究生创新基金资助项目(DHYC-201927)

Crop Leaf Diseases Recognition: A Generative Adversarial Network Based Approach

  1. (School of Information Engineering, East China University of Technology, Nanchang 330013, China)
  • Online:2020-12-03 Published:2020-12-03

摘要: 当人们使用深度神经网络对图像进行分类时,通常需要大量的训练样本。然而,在实际工作中很难获得足够多的样本来保证神经网络的训练。为了解决这一问题,本文提出一种基于生成式对抗网络的识别方法。其主要思想是通过对现有的GAN网络模型进行改造后训练一个样本生成模型,然后利用神经网络对生成模型生成的数据集进行识别,最后利用迁移学习方法对具有真实数据的神经网络进行微调。为了验证该方法的有效性,本文使用5种作物的叶片进行验证(每个样本500片),其对植物叶片的有无病害识别精度可达90%以上。实验结果表明该方法能在少量样本时提高叶片的识别精度,具有很强的通用性。

关键词: 农作物叶片, 病害识别, 生成对抗网络, 神经网络, 微调

Abstract: When people use deep neural networks to classify images, they usually need a large number of training samples. However, it is difficult to obtain enough samples to ensure the training of neural network in practice. In order to solve this problem, this paper proposes an identification method based on generative adversarial network. The main idea is to train a sample generation model after modification of the existing GAN network model, then use neural network to identify the data set generated by the generation model, and finally use transfer learning method to fine-tune the neural network with real data. In order to verify the effectiveness of this method, five crop leaves (500 pieces per sample) are used for validation, the identification accuracy of plant leaves is more than 90%. The experimental results show that this method can improve the identification accuracy of the blade with a small number of samples and has strong universality.

Key words: crop leaf, diseases recognition, generative adversarial network, neural network, fine-tune