吉林大学学报(理学版) ›› 2022, Vol. 60 ›› Issue (4): 906-910.

• • 上一篇    下一篇

针对苹果树叶病害图像分类的小样本学习方法

李蛟1, 王紫薇2, 范丽丽3, 赵宏伟3   

  1. 1. 吉林大学 图书馆, 长春 130012; 2. 吉林省商务信息中心, 长春 130061;3. 吉林大学 计算机科学与技术学院, 长春 130012
  • 收稿日期:2021-10-20 出版日期:2022-07-26 发布日期:2022-07-26
  • 通讯作者: 王紫薇 E-mail:wzw@ji.gov.cn

Few-Shot Learning Method for Image Classification of Apple Leaf Diseases

LI Jiao1, WANG Ziwei2, FAN Lili3, ZHAO Hongwei3   

  1. 1. Library of Jilin University, Changchun 130012, China;2. Center of Jilin Business Information, Changchun 130061, China;
    3. College of Computer Science and Technology, Jilin University, Changchun 130012, China
  • Received:2021-10-20 Online:2022-07-26 Published:2022-07-26

摘要: 针对苹果树叶病害样本数量少且缺乏足够的标签, 导致苹果树叶病害早期发现分类困难的问题, 提出一种苹果树叶病害图像分类的小样本学习方法. 先根据图像间的特征向量距离和样本点密度, 找出样本中的离群因子, 将离群因子剔除后, 求取嵌入空间中支持集的平均值, 再根据查找样本与该值的关系进行分类. 实验结果表明, 该小样本学习方法能明显提高模型的分类性能和收敛速度, 分类准确率较高, 平均分类准确率达97.62%, 对苹果树叶锈病、黑星病、混合病害、健康树叶4类的分类准确率分别达98.01%,97.32%,96.30%,98.85%, 且对样本不平衡、 背景不均匀等数据集有较强的鲁棒性.

关键词: 苹果树叶病害, 图像分类, 小样本学习, 离群点剔除

Abstract: Aiming at the problem of small number of  samples of apple leaf disease and the lack of sufficient labels, which led to the difficulty of the early detection and classification of apple leaf diseases, we proposed a few-shot learning method for image classification of apple leaf diseases. First, according to the distance of feature vectors  between the images and the density of sample points, we found out the outlier factors in the samples, after the outlier factors were eliminated, the average value of the support set in the embedding space was calculated, and then the classification was performed according to the relationship between the search sample and the value. The experimental results show that the proposed few-shot learning method can significantly improve the classification performance and convergence speed of the model, and has high classification accuracy, its  average classification accuracy is 97.62%, and the classification accuracy of apple rust, scab, mixed diseases and healthy leaves is 98.01%, 97.32%, 96.30%, 98.85%, respectively. It has  strong robustness to data sets such as sample imbalance and uneven background, etc.

Key words: apple leaf disease, image classification, few-shot learning, outlier removal

中图分类号: 

  • TP391