Advancing common bean (Phaseolus vulgaris L.) disease detection with YOLO driven deep learning to enhance agricultural AI
Daniela Gomez,Michael Gomez Selvaraj,Jorge Casas,Kavino Mathiyazhagan,Michael Rodriguez,Teshale Assefa,Anna Mlaki,Goodluck Nyakunga,Fred Kato,Clare Mukankusi,Ellena Girma,Gloria Mosquera,Victoria Arredondo,Ernesto Espitia
DOI: https://doi.org/10.1038/s41598-024-66281-w
2024-07-06
Abstract:Common beans (CB), a vital source for high protein content, plays a crucial role in ensuring both nutrition and economic stability in diverse communities, particularly in Africa and Latin America. However, CB cultivation poses a significant threat to diseases that can drastically reduce yield and quality. Detecting these diseases solely based on visual symptoms is challenging, due to the variability across different pathogens and similar symptoms caused by distinct pathogens, further complicating the detection process. Traditional methods relying solely on farmers' ability to detect diseases is inadequate, and while engaging expert pathologists and advanced laboratories is necessary, it can also be resource intensive. To address this challenge, we present a AI-driven system for rapid and cost-effective CB disease detection, leveraging state-of-the-art deep learning and object detection technologies. We utilized an extensive image dataset collected from disease hotspots in Africa and Colombia, focusing on five major diseases: Angular Leaf Spot (ALS), Common Bacterial Blight (CBB), Common Bean Mosaic Virus (CBMV), Bean Rust, and Anthracnose, covering both leaf and pod samples in real-field settings. However, pod images are only available for Angular Leaf Spot disease. The study employed data augmentation techniques and annotation at both whole and micro levels for comprehensive analysis. To train the model, we utilized three advanced YOLO architectures: YOLOv7, YOLOv8, and YOLO-NAS. Particularly for whole leaf annotations, the YOLO-NAS model achieves the highest mAP value of up to 97.9% and a recall of 98.8%, indicating superior detection accuracy. In contrast, for whole pod disease detection, YOLOv7 and YOLOv8 outperformed YOLO-NAS, with mAP values exceeding 95% and 93% recall. However, micro annotation consistently yields lower performance than whole annotation across all disease classes and plant parts, as examined by all YOLO models, highlighting an unexpected discrepancy in detection accuracy. Furthermore, we successfully deployed YOLO-NAS annotation models into an Android app, validating their effectiveness on unseen data from disease hotspots with high classification accuracy (90%). This accomplishment showcases the integration of deep learning into our production pipeline, a process known as DLOps. This innovative approach significantly reduces diagnosis time, enabling farmers to take prompt management interventions. The potential benefits extend beyond rapid diagnosis serving as an early warning system to enhance common bean productivity and quality.