Research on grape leaf classification based on optimized densenet201 model

Jian Huang

PLOS ONE, 2025, vol. 20, issue 10, 1-21

Abstract: In the realm of plant classification, the classification of grape leaf varieties has long presented a complex challenge. Aiming to enhance the accuracy and generalization ability of grape leaf variety classification, this study proposes a novel approach that employs an optimized Densenet201 model for grape leaf classification. Initially, grape leaf images from five distinct varieties were meticulously collected to construct a comprehensive grape leaf dataset. To augment the diversity of the dataset, the parameters of data augmentation were delicately adjusted, with an increase in the rotation range, translation range, and so on. Subsequently, BatchNormalization and GlobalAveragePooling2D layers were incorporated to achieve feature normalization and pooling. Simultaneously, the parameters of the Dropout layer were optimized to effectively mitigate the issue of overfitting. Additionally, the number of neurons and layers in the Dense layer were varied to explore diverse network structures and pursue superior performance. Moreover, the parameters of the Adam optimizer were meticulously tuned to attain the optimal performance, and the model’s performance was further enhanced by extracting image features. The experimental results demonstrate that, in comparison with the densenet121, densenet169, resnet50, and densenet201 models, the optimized Densenet201 model showcases outstanding performance in grape leaf variety classification, remarkably improving the classification accuracy and generalization ability. This research provides a more efficient method for grape leaf variety classification.

Date: 2025
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0334877 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 34877&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0334877

DOI: 10.1371/journal.pone.0334877

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().