首页|基于对比学习的植物叶片病害识别

基于对比学习的植物叶片病害识别

扫码查看
目前,基于图像处理的植物病害识别多依赖于人工标注的卷积神经网络。本文基于自监督对比学习不依赖标签和大量数据即可实现独自学习的优势,研究了MoCo-v2、DeepCluster-v2、SwAV、BYOL这 4 种对比学习方法对植物叶片病害的识别效果,通过设置不同试验条件,使用PlantVillage开源数据集和自建的棉花病害数据集分别测试4 种对比学习方法所训练的ResNet50 编码器在Linear和Finetune两种模式下的病害识别效果,评估对比学习方法在植物叶片病害识别上的可行性。结果表明,在PlantVillage数据集中,Finetune模式的平均准确率略高于Linear,4 种方法训练的编码器的平均识别准确率最高达 99。83%,其中,DeepCluster-v2、BYOL在Finetune模式下的识别率最高,均为99。87%。在棉花病害数据集上,Finetune模式的效果略逊于Linear,DeepCluster-v2 在Linear模式下获得最高识别准确率(98。86%)。整体来看,基于对比学习方法的病害识别率优于有监督模型的学习效果,在植物叶片病害识别领域展现出良好的应用前景。
Plant leaf disease identification based on contrastive learning
At present,the recognition of plant disease via image processing mostly relies on the manually labeled convolutional neural network.However,the self-monitoring contrastive learning could achieve independent learning without relying on labels and large amounts of data.In view of this advantage,the effect of four contrastive learning methods,MoCo-v2,DeepCluster-v2,SwAV and BYOL,on the identification of plant leaf diseases were compared by setting different experimental conditions on the open-source dataset of PlantVillage and the self-built cotton disease dataset.The ResNet50 encoder trained by four contrastive learning methods was tested for disease identification both under Linear and Finetune modes,and the feasibility of the contrastive learning methods in identifying plant leaf dis-eases was evaluated.It was shown that the average accuracy under Finetune mode on the PlantVillage dataset was higher than that under Linear mode,and the highest identificaiton accuracy of the encoders trained by the four meth-ods reached 99.83%.DeepCluster-v2 and BYOL had the highest identification rate under Finetune mode,both of which were 99.87%.On the self-built cotton disease dataset,the performance under Finetune mode was poorer than that under Linear mode,and the highest idenficaiton accuracy of DeepCluster-v2 under Linear mode was 98.86%.Overall,the disease identification rate based on contrastive learning method was superior to the supervised models,demonstrating good application prospects in plant leaf disease identification.

contrastive learningdisease identificationimage processing

杨新宇、冯全、张建华、杨森

展开 >

甘肃农业大学 机电工程学院,甘肃 兰州 730070

中国农业科学院 农业信息研究所,北京 100081

对比学习 病害识别 图像处理

国家自然科学基金国家自然科学基金国家自然科学基金甘肃省高等学校产业支撑计划

3216042131971792322016632021CYZC-57

2024

浙江农业学报
浙江省农业科学院 浙江省农学会

浙江农业学报

CSTPCD北大核心
影响因子:0.765
ISSN:1004-1524
年,卷(期):2024.36(1)
  • 19