Skip to main content

ORIGINAL RESEARCH article

Front. Plant Sci.
Sec. Sustainable and Intelligent Phytoprotection
Volume 15 - 2024 | doi: 10.3389/fpls.2024.1475282
This article is part of the Research Topic Innovative Approaches to Agricultural Plant Disease Identification: Integrating Deep Learning into Traditional Methods View all 12 articles

DINOV2-FCS: a model for fruit leaf disease classification and severity prediction

Provisionally accepted
Chunhui Bai Chunhui Bai 1,2,3Lilian Zhang Lilian Zhang 1,2,3*Lutao Gao Lutao Gao 1,2,3*Lin Peng Lin Peng 1,2,3*Peishan Li Peishan Li 1,2,3Linnan Yang Linnan Yang 1,2,3*
  • 1 College of Big Data, Yunnan Agricultural University, Kunming, China
  • 2 Yunnan Engineering Technology Research Center of Agricultural Big Data, Kunming, China
  • 3 Yunnan Engineering Research Center for Big Data Intelligent Information Processing of Green Agricultural Products, Kunming, China

The final, formatted version of the article will be published soon.

    The assessment of the severity of fruit disease is crucial for the optimization of fruit production. By quantifying the percentage of leaf disease, an effective approach to determining the severity of the disease is available. However, the current prediction of disease degree by machine learning methods still faces challenges, including suboptimal accuracy and limited generalizability. In light of the growing application of large model technology across a range of fields, this study draws upon the DINOV2 visual large vision model backbone network to construct the DINOV2-Fruit Leaf Classification and Segmentation Model (DINOV2-FCS), a model designed for the classification and severity prediction of diverse fruit leaf diseases. DINOV2-FCS employs the DINOv2-B (distilled) backbone feature extraction network to enhance the extraction of features from fruit disease leaf images. In fruit leaf disease classification, for the problem that leaf spots of different diseases have great similarity, we have proposed Class-Patch Feature Fusion Module (C-PFFM), which integrates the local detailed feature information of the spots and the global feature information of the class markers. For the problem that the model ignores the fine spots in the segmentation process, we propose Explicit Feature Fusion Architecture (EFFA) and Alterable Kernel Atrous Spatial Pyramid Pooling (AKASPP), which improve the segmentation effect of the model. To verify the accuracy and generalizability of the model, two sets of experiments were conducted. First, the labeled leaf disease dataset of five fruits was randomly divided. The trained model exhibited an accuracy of 99.67% in disease classification, an mIoU of 90.29%, and an accuracy of 95.68% in disease severity classification. These results demonstrate superior performance compared to other state-of-the-art models. In the generalizability experiment, four disease data sets were used for training and one for testing. The mIoU of the trained model reached 83.95%, and the accuracy of disease severity grading was 95.24%. The strong generalization ability of the model was verified. This study provides a new method for leaf disease classification and leaf disease severity prediction for a variety of fruits. Code is available at https://github.com/BaiChunhui2001/DINOV2-FCS.

    Keywords: DINOV2, deep learning, fruit disease recognition, Semantic segmentation, smart agriculture

    Received: 03 Aug 2024; Accepted: 20 Nov 2024.

    Copyright: © 2024 Bai, Zhang, Gao, Peng, Li and Yang. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

    * Correspondence:
    Lilian Zhang, College of Big Data, Yunnan Agricultural University, Kunming, China
    Lutao Gao, College of Big Data, Yunnan Agricultural University, Kunming, China
    Lin Peng, College of Big Data, Yunnan Agricultural University, Kunming, China
    Linnan Yang, College of Big Data, Yunnan Agricultural University, Kunming, China

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.