Home / Current Issue / Paper 1717169
Machine Learning and Deep Learning for Plant Disease Classification and Detection
Subject area: Science,Engineering and Technology · Area of research: Artificial Intelligence
DOI: https://doi.org/10.64388/IREV9I10-1717169
Abstract
Plant diseases cause significant losses in global agricultural output, and early, accurate identification remains a major challenge. This paper presents a unified deep learning pipeline that simultaneously localises and classifies plant diseases from leaf images. A YOLOv5s model performs region-level detection, while a transfer-learning-based ResNet50 classifier identifies the specific disease category. The system is trained on the PlantVillage dataset, which contains 54,306 images spanning 38 disease classes across 14 crop species. Preprocessing includes image resizing, normalisation, and a combination of standard and advanced augmentation strategies, namely CutMix and MixUp. ResNet50 is fine-tuned in two phases with Cosine Learning Rate decay, and Test Time Augmentation (TTA) is applied at inference for additional accuracy gains. The ResNet50 classifier achieves 99.1% validation accuracy with TTA, while the YOLOv5s detection model reaches a mAP@0.50 of 0.856. The macro-averaged precision, recall, and F1-score stand at 0.974, 0.968, and 0.971, respectively. The integrated pipeline demonstrates reliable end-to-end performance with 97.6% accuracy on 200 held-out images, offering a practical and scalable solution for precision agriculture.
Keywords
Plant Disease Detection, Deep Learning, ResNet50, YOLOv5, Transfer Learning, Precision Agriculture, CutMix, MixUp, Test Time Augmentation, PlantVillage.
References
[1] V. Balafas, E. Karantoumanis, M. Louta, and N. Ploskas, “Machine Learning and Deep Learning for Plant Disease Classification and Detection,” IEEE Access, vol. 11, pp. 114352–114377, 2023.
[2] S. P. Mohanty, D. P. Hughes, and M. Salathé, “Using Deep Learning for Image-Based Plant Disease Detection,” Front. Plant Sci., vol. 7, Art. 1419, 2016.
[3] K. He, X. Zhang, S. Ren, and J. Sun, “Deep Residual Learning for Image Recognition,” in Proc. IEEE CVPR, 2016, pp. 770–778.
[4] J. Redmon, S. Divvala, R. Girshick, and A. Farhadi, “You Only Look Once: Unified, Real-Time Object Detection,” in Proc. IEEE CVPR, 2016, pp. 779–788.
[5] G. Jocher, YOLOv5 by Ultralytics, v3.1, 2022. [Online]. Available: https://github.com/ultralytics/yolov5
[6] D. P. Hughes and M. Salathé, “An Open Access Repository of Images on Plant Health to Enable the Development of Mobile Disease Diagnostics,” arXiv:1511.08060, 2015.
[7] D. Singh et al., “PlantDoc: A Dataset for Visual Plant Disease Detection,” in Proc. ACM IKDD CoDS-COMAD, 2020, pp. 249–253.
[8] K. Simonyan and A. Zisserman, “Very Deep Convolutional Networks for Large-Scale Image Recognition,” arXiv:1409.1556, 2014.
[9] D. P. Kingma and J. Ba, “Adam: A Method for Stochastic Optimization,” arXiv:1412.6980, 2014.
[10] S. Yun et al., “CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Features,” in Proc. IEEE ICCV, 2019, pp. 6023–6032.
[11] H. Zhang et al., “MixUp: Beyond Empirical Risk Minimization,” in ICLR, 2018.
[12] A. Fuentes, S. Yoon, S. Kim, and D. Park, “A Robust Deep-Learning-Based Detector for Real-Time Tomato Plant Diseases and Pests Recognition,” Sensors, vol. 17, no. 9, Art. 2022, 2017.
[13] S. Ren, K. He, R. Girshick, and J. Sun, “Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks,” in NeurIPS, 2015, pp. 91–99.
[14] W. Liu et al., “SSD: Single Shot MultiBox Detector,” in Proc. ECCV, 2016, pp. 21–37.
[15] K. P. Ferentinos, “Deep Learning Models for Plant Disease Detection and Diagnosis,” Comput. Electron. Agric., vol. 145, pp. 311–318, 2018.
How to cite this paper
@article{1717169,
author = {N. Sri Janaki Ram, N. Venkata Sai Gopi Chandu, S. Ajay, Reddy Veeramohanarao},
title = {Machine Learning and Deep Learning for Plant Disease Classification and Detection},
journal = {Iconic Research And Engineering Journals},
year = {2026},
volume = {9},
number = {10},
pages = {3382-3385},
issn = {2456-8880},
url = {https://www.irejournals.com/formatedpaper/1717169.pdf},
abstract = {Plant diseases cause significant losses in global agricultural output, and early, accurate identification remains a major challenge. This paper presents a unified deep learning pipeline that simultaneously localises and classifies plant diseases from leaf images. A YOLOv5s model performs region-level detection, while a transfer-learning-based ResNet50 classifier identifies the specific disease category. The system is trained on the PlantVillage dataset, which contains 54,306 images spanning 38 disease classes across 14 crop species. Preprocessing includes image resizing, normalisation, and a combination of standard and advanced augmentation strategies, namely CutMix and MixUp. ResNet50 is fine-tuned in two phases with Cosine Learning Rate decay, and Test Time Augmentation (TTA) is applied at inference for additional accuracy gains. The ResNet50 classifier achieves 99.1% validation accuracy with TTA, while the YOLOv5s detection model reaches a mAP@0.50 of 0.856. The macro-averaged precision, recall, and F1-score stand at 0.974, 0.968, and 0.971, respectively. The integrated pipeline demonstrates reliable end-to-end performance with 97.6% accuracy on 200 held-out images, offering a practical and scalable solution for precision agriculture.},
keywords = {Plant Disease Detection, Deep Learning, ResNet50, YOLOv5, Transfer Learning, Precision Agriculture, CutMix, MixUp, Test Time Augmentation, PlantVillage.},
month = {April},
doi = {https://doi.org/10.64388/IREV9I10-1717169}
}