基于双节点-双边图神经网络的茶叶病害分类方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

安徽省中央引导地方科技发展专项(202107d06020001)和国家自然科学基金项目(32372632)


Tea Disease Classification Method Based on Graph Neural Network with Dual Nodes-Dual Edges
Author:
Affiliation:

Fund Project:

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

    传统茶叶病害分类主要依赖人工方法,此类方法费工费时,同时茶叶病害样本较少使得现有的机器学习方法的模型训练不充分,病害分类准确率不够高。针对茶炭疽病、茶黑煤病、茶饼病和茶白星病4类病害,提出一种基于双节点-双边图神经网络的茶叶病害分类方法。首先通过两分支卷积神经网络提取RGB茶叶病害特征和灰度茶叶病害特征,两分支均采用ResNet12作为骨干网络,参数独立不共享,两类特征作为图神经网络的两个子节点,以获得不同域样本所包含的病害信息;其次构建相对度量边和相似性边两类边,从而强化节点对相邻节点所含病害特征的聚合能力。最后,经过双节点特征和双边特征更新模块,实现双节点和双边交替更新,提高边特征对节点距离度量的准确性,从而实现训练样本较少条件下对茶叶病害的准确分类。本文方法和小样本学习方法进行了对比实验,结果表明,本文方法获得更高的准确率,在miniImageNet和PlantVillage数据集上5way-1shot的准确率分别达到69.30%和88.42%,5way-5shot准确率分别为82.48%和93.04%。同时在茶叶数据集TeaD-5上5way-1shot和5way-5shot准确率分别达到84.74%和86.34%。

    Abstract:

    The classification of traditional tea diseases mainly relies on manual categorization. Such methods are labor-intensive and time-consuming.Furthermore, insufficient availability of tea disease samples hampers the adequate training of existing machine learning models, resulting in decreased accuracy in disease classification. To address this problem, a tea disease classification method was proposed for four types of tea diseases, including tea anthracnose, tea black rot, and others. This method was based on a dual node-dual edge graph neural network.Firstly, RGB tea disease features and grayscale tea disease features were extracted by using two branches of convolutional neural networks, both branches employed ResNet12 as the backbone network, with independent parameters.The two types of features acted as two sub-nodes within the graph neural network, aiming to obtain disease information from different domains. Secondly, two types of edges, including relative metric edges and similarity edges, were created to improve the aggregation capability of disease features from neighboring nodes.Finally, with the dual node and dual edge feature updating modules, a dual-node and dual-edge alternate updating process was achieved. This process aimed to enhance the accuracy of edge features in measuring node distances. This resulted in achieving accurate classification of tea diseases, even when training samples were limited. Comparative experiments were conducted between the proposed methods, which were based on small-sample learning method. The results indicated that the proposed method achieved superior accuracy in tea disease classification. Specifically, on the miniImageNet and PlantVillage datasets, the proposed method achieved the accuracy of 69.30% and 88.42% in the 5way-1shot, respectively. In the 5way-5shot, the accuracy was improved to 82.48% and 93.04% on the miniImageNet and PlantVillage datasets. Furthermore, on the TeaD-5 tea dataset, the accuracy of the proposed method reached 84.74% in the 5way-1shot and 86.34% in the 5way-5shot.

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

张艳,车迅,汪芃,汪玉凤,胡根生.基于双节点-双边图神经网络的茶叶病害分类方法[J].农业机械学报,2024,55(3):252-262. ZHANG Yan, CHE Xun, WANG Peng, WANG Yufeng, HU Gensheng. Tea Disease Classification Method Based on Graph Neural Network with Dual Nodes-Dual Edges[J]. Transactions of the Chinese Society for Agricultural Machinery,2024,55(3):252-262.

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