Deep Learning for Tomato Disease Detection with YOLOv8
Received: 13 February 2024 | Revised: 28 February 2024 | Accepted: 29 February 2024 | Online: 7 March 2024
Corresponding author: Hafedh Mahmoud Zayani
Abstract
Tomato production plays a crucial role in Saudi Arabia, with significant yield variations due to factors such as diseases. While automation offers promising solutions, accurate disease detection remains a challenge. This study proposes a deep learning approach based on the YOLOv8 algorithm for automated tomato disease detection. Augmenting an existing Roboflow dataset, the model achieved an overall accuracy of 66.67%. However, class-specific performance varies, highlighting challenges in differentiating certain diseases. Further research is suggested, focusing on data balancing, exploring alternative architectures, and adopting disease-specific metrics. This work lays the foundation for a robust disease detection system to improve crop yields, quality, and sustainable agriculture in Saudi Arabia.
Keywords:
disease detection tomato, deep learning, YOLOv8Downloads
References
J. M. Costa and E. Heuvelink, "The global tomato industry," in Tomatoes, CABI, 2018, pp. 1–26. DOI: https://doi.org/10.1079/9781780641935.0001
"Huge increase in tomato production in Saudi Arabia due to targeted watering," Jan. 20, 2020. https://www.wur.nl/en/research-results/research-institutes/plant-research/greenhouse-horticulture/show-greenhouse/huge-increase-in-tomato-production-in-saudi-arabia-due-to-targeted-watering.htm.
I. Tiwari et al., "Post-harvest practices and loss assessment in tomato (Solanum lycopersicum L.) in Kathmandu, Nepal," Journal of Agriculture and Natural Resources, vol. 3, no. 2, pp. 335–352, Oct. 2020. DOI: https://doi.org/10.3126/janr.v3i2.32545
M. Benavides, M. Cantón-Garbín, J. A. Sánchez-Molina, and F. Rodríguez, "Automatic Tomato and Peduncle Location System Based on Computer Vision for Use in Robotized Harvesting," Applied Sciences, vol. 10, no. 17, 2020. DOI: https://doi.org/10.3390/app10175887
Y. Bai, S. Mao, J. Zhou, and B. Zhang, "Clustered tomato detection and picking point location using machine learning-aided image analysis for automatic robotic harvesting," Precision Agriculture, vol. 24, no. 2, pp. 727–743, Apr. 2023. DOI: https://doi.org/10.1007/s11119-022-09972-6
Y. Gulzar, "Fruit Image Classification Model Based on MobileNetV2 with Deep Transfer Learning Technique," Sustainability, vol. 15, no. 3, 2023. DOI: https://doi.org/10.3390/su15031906
M. Afonso et al., "Tomato Fruit Detection and Counting in Greenhouses Using Deep Learning," Frontiers in Plant Science, vol. 11, 2020. DOI: https://doi.org/10.3389/fpls.2020.571299
G. Moreira, S. A. Magalhães, T. Pinho, F. N. dos Santos, and M. Cunha, "Benchmark of Deep Learning and a Proposed HSV Colour Space Models for the Detection and Classification of Greenhouse Tomato," Agronomy, vol. 12, no. 2, Feb. 2022, Art. no. 356. DOI: https://doi.org/10.3390/agronomy12020356
Y. Mu, T. S. Chen, S. Ninomiya, and W. Guo, "Intact Detection of Highly Occluded Immature Tomatoes on Plants Using Deep Learning Techniques," Sensors, vol. 20, no. 10, 2020. DOI: https://doi.org/10.3390/s20102984
J. Wu, Z. Kuang, L. Wang, W. Zhang, and G. Wu, "Context-Aware RCNN: A Baseline for Action Detection in Videos," in Computer Vision – ECCV 2020, Glascow, UK, Aug. 2020, pp. 440–456. DOI: https://doi.org/10.1007/978-3-030-58595-2_27
F. Zeng, Y. Liu, Y. Ye, J. Zhou, and X. Liu, "A detection method of Edge Coherent Mode based on improved SSD," Fusion Engineering and Design, vol. 179, Jun. 2022, Art. no. 113141. DOI: https://doi.org/10.1016/j.fusengdes.2022.113141
H. Peng et al., "General improved SSD model for picking object recognition of multiple fruits in natural environment.," Transactions of the Chinese Society of Agricultural Engineering, vol. 34, no. 16, pp. 155–162, 2018.
P. Jiang, D. Ergu, F. Liu, Y. Cai, and B. Ma, "A Review of Yolo Algorithm Developments," Procedia Computer Science, vol. 199, pp. 1066–1073, Jan. 2022. DOI: https://doi.org/10.1016/j.procs.2022.01.135
G. Liu, J. C. Nouaze, P. L. Touko Mbouembe, and J. H. Kim, "YOLO-Tomato: A Robust Algorithm for Tomato Detection Based on YOLOv3," Sensors, vol. 20, no. 7, 2020. DOI: https://doi.org/10.3390/s20072145
Y. Yang, J. Li, J. Nie, S. Yang, and J. Tang, "Cotton Stubble Detection Based on Improved YOLOv3," Agronomy, vol. 13, no. 5, 2023. DOI: https://doi.org/10.3390/agronomy13051271
R. Gai, N. Chen, and H. Yuan, "A detection algorithm for cherry fruits based on the improved YOLO-v4 model," Neural Computing and Applications, vol. 35, no. 19, pp. 13895–13906, Jul. 2023. DOI: https://doi.org/10.1007/s00521-021-06029-z
R. Li, Z. Ji, S. Hu, X. Huang, J. Yang, and W. Li, "Tomato Maturity Recognition Model Based on Improved YOLOv5 in Greenhouse," Agronomy, vol. 13, no. 2, 2023. DOI: https://doi.org/10.3390/agronomy13020603
R. Rajamohanan and B. C. Latha, "An Optimized YOLO v5 Model for Tomato Leaf Disease Classification with Field Dataset," Engineering, Technology & Applied Science Research, vol. 13, no. 6, pp. 12033–12038, Dec. 2023. DOI: https://doi.org/10.48084/etasr.6377
T. Saidani, "Deep Learning Approach: YOLOv5-based Custom Object Detection," Engineering, Technology & Applied Science Research, vol. 13, no. 6, pp. 12158–12163, Dec. 2023. DOI: https://doi.org/10.48084/etasr.6397
T. Saidani, R. Ghodhbani, A. Alhomoud, A. Alshammari, H. Zayani, and M. B. Ammar, "Hardware Acceleration for Object Detection using YOLOv5 Deep Learning Algorithm on Xilinx Zynq FPGA Platform," Engineering, Technology & Applied Science Research, vol. 14, no. 1, pp. 13066–13071, Feb. 2024. DOI: https://doi.org/10.48084/etasr.6761
J. Zhou, Y. Zhang, and J. Wang, "RDE-YOLOv7: An Improved Model Based on YOLOv7 for Better Performance in Detecting Dragon Fruits," Agronomy, vol. 13, no. 4, 2023. DOI: https://doi.org/10.3390/agronomy13041042
G. Yang, J. Wang, Z. Nie, H. Yang, and S. Yu, "A Lightweight YOLOv8 Tomato Detection Algorithm Combining Feature Enhancement and Attention," Agronomy, vol. 13, no. 7, 2023. DOI: https://doi.org/10.3390/agronomy13071824
B. N. E. Nyarko, W. Bin, Z. Jinzhi, and J. Odoom, "Tomato fruit disease detection based on improved single shot detection algorithm," Journal of Plant Protection Research, vol. 63, no. 4, pp. 405–417, Oct. 2023.
A. Santosh, S. Bikesh, B. Bibek, and K. K. C. Saban, "Tomato Plant Diseases Detection System Using Image Processing," in 1st KEC Conference Proceedings, Sep. 2018, pp. 81–86.
Q. H. Phan, V. T. Nguyen, C. H. Lien, T. P. Duong, M. T. K. Hou, and N. B. Le, "Classification of Tomato Fruit Using Yolov5 and Convolutional Neural Network Models," Plants, vol. 12, no. 4, 2023. DOI: https://doi.org/10.3390/plants12040790
S. N. Appe, A. G, and B. Gn, "CAM-YOLO: tomato detection and classification based on improved YOLOv5 using combining attention mechanism," PeerJ Computer Science, vol. 9, Jul. 2023, Art. no. e1463. DOI: https://doi.org/10.7717/peerj-cs.1463
“balanceddata Computer Vision Project.” Roboflow Universe, Oct. 2023, [Online]. Available: https://universe.roboflow.com/research-proj-mgap9/balanceddata-y4ox0.
S. Mukherjee, "YOLOv8: Pioneering Breakthroughs in Object Detection Technology," Paperspace Blog, Nov. 03, 2023. https://blog.paperspace.com/yolov8-a-revolutionary-advancement-in-object-detection-2/.
Z. Zhang, "Drone-YOLO: An Efficient Neural Network Method for Target Detection in Drone Images," Drones, vol. 7, no. 8, 2023. DOI: https://doi.org/10.3390/drones7080526
Downloads
How to Cite
License
Copyright (c) 2024 Hafedh Zayani, Ikhlass Ammar, Refka Ghodhbani, Albia Maqbool, Taoufik Saidani, Jihane Ben Slimane, Amani Kachoukh, Marouan Kouki, Mohamed Kallel, Amjad A. Alsuwaylimi, Sami Mohammed Alenezi
This work is licensed under a Creative Commons Attribution 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain the copyright and grant the journal the right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) after its publication in ETASR with an acknowledgement of its initial publication in this journal.