基于迁移学习的棉花叶部病虫害图像识别. (Chinese)

Image recognition of cotton leaf diseases and pests based on transfer learning. (English)

The whole growth cycle of cotton is attacked by more 40 kinds of diseases and insect pests, which seriously affect its yield. Therefore, it is very important to identify the types of cotton diseases quickly and accurately and to control them timely and accurately to avoid the further spread of the disease and improve the yield of cotton. In view of the low accuracy of traditional cotton pest identification and the need for manual image feature extraction, a convolution neural network method is proposed to classify cotton leaf pests. Because it is difficult to obtain large data sets in the field of agriculture. Transfer learning and data enhancement are used to deal with small data sets. The network structure of 5 convolution layers, 2 full connection layers, and 1 Softmax classification layer were built based on the AlexNet model. The model was used to classify six diseases and insect pests in cotton leaves. The experiment was divided into 2 parts. The first part used to a PlantVillage big data set to learn the pre-training model on the build model as the feature extractor to save the model. Then, we used the transfer learning method of model transfer and fine-tuning parameters to train our model on the original cotton pest data set. The original cotton pest data set collected in the experiment was divided into training according to the proportion of 6:2:2 training set, verification set, and test set. The average test accuracy was 93.50% through 3 transfer learning training mechanisms (Freezing C1, C2, C3, C4, C5; freeze C1, C2, C3, C4, C5, and F6; freeze C1, C2, C3, C4, C5, and F6, F7). The second part used data enhancement technology to expand the original cotton pest data set to get a new set and then repeated the first part of the experiment with the new data set instead of the original data set. The cotton pest data was still divided in the proportion of 6:2:2, and then used two kinds of training mechanisms to transfer and learn (freeze C1, C2, C3, C4, C5; freeze C1, C2, C3, C4, C5, and F6), and the final average test set accuracy was 97.16%. Under the same experimental conditions, the accuracy of this model was much higher than that of traditional image classification methods, such as SVM and BP neural network. The experimental results of the deep convolution model VGG-19 and the GoogLeNet Inception v2 model showed that the improved model could converge faster and higher classification accuracy. The experimental results showed that the knowledge learned from the big data set of PlantVillage could be transferred to the target area (cotton pests data set) through the transfer learning, and the solution the problem of small data set could also make the model converge quickly. The method of data enhancement method could effectively alleviate the overfitting problem. This study had a good recognition rate for the pests of cotton leaves and provides a reference for the development of crop pest identification technology. [ABSTRACT FROM AUTHOR]

针对传统图像识别方法准确率低、手工提取特征等问题，该研究以棉花叶部病虫害图像为研究对象，利用迁移 学习算法并辅以数据增强技术，实现棉花叶部病虫害图像准确分类。首先改进 AlexNet 模型，利用 PlantVillage 大数据集 训练取得预训练模型，在预训练模型上使用棉花病虫害数据微调参数，得到平均测试准确率为 93.50%；然后使用数据增强 技术扩充原始数据集，在预训练模型上再训练，得到终平均测试准确率为 97.16%。相同试验条件下，该研究方法较支持 向量机（Support Vector Machine，SVM）和 BP（Back Propagation，BP）神经网络以及深度卷积模型（VGG-19 和GoogLeNet Inception v2）分类效果更好。试验结果表明，通过迁移学习能把从源领域（PlantVillage 数据集）学习到的知识迁移到目标 领域（棉花病虫害数据集），数据增强技术能有效缓解过拟合。该研究为农作物病虫害识别技术的发展提供了参考。 [ABSTRACT FROM AUTHOR]

Copyright of Transactions of the Chinese Society of Agricultural Engineering is the property of Chinese Society of Agricultural Engineering and its content may not be copied or emailed to multiple sites or posted to a listserv without the copyright holder's express written permission. However, users may print, download, or email articles for individual use. This abstract may be abridged. No warranty is given about the accuracy of the copy. Users should refer to the original published version of the material for the full abstract. (Copyright applies to all Abstracts.)