YOLOv8-GABNet: An Enhanced Lightweight Network for the High-Precision Recognition of Citrus Diseases and Nutrient Deficiencies

Qiufang Dai, Yungao Xiao, Shilei Lv, Shuran Song, Xiuyun Xue, Shiyao Liang, Ying Huang and Zhen Li ()
Additional contact information
Qiufang Dai: College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China
Yungao Xiao: College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China
Shilei Lv: College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China
Shuran Song: College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China
Xiuyun Xue: College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China
Shiyao Liang: College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China
Ying Huang: College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China
Zhen Li: College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China

Agriculture, 2024, vol. 14, issue 11, 1-18

Abstract: Existing deep learning models for detecting citrus diseases and nutritional deficiencies grapple with issues related to recognition accuracy, complex backgrounds, occlusions, and the need for lightweight architecture. In response, we developed an improved YOLOv8-GABNet model designed specifically for citrus disease and nutritional deficiency detection, which effectively addresses these challenges. This model incorporates several key enhancements: A lightweight ADown subsampled convolutional block is utilized to reduce both the model’s parameter count and its computational demands, replacing the traditional convolutional module. Additionally, a weighted Bidirectional Feature Pyramid Network (BiFPN) supersedes the original feature fusion network, enhancing the model’s ability to manage complex backgrounds and achieve multiscale feature extraction and integration. Furthermore, we introduced important features through the Global to Local Spatial Aggregation module (GLSA), focusing on crucial image details to enhance both the accuracy and robustness of the model. This study processed the collected images, resulting in a dataset of 1102 images. Using LabelImg, bounding boxes were applied to annotate leaves affected by diseases. The dataset was constructed to include three types of citrus diseases—anthracnose, canker, and yellow vein disease—as well as two types of nutritional deficiencies, namely magnesium deficiency and manganese deficiency. This dataset was expanded to 9918 images through data augmentation and was used for experimental validation. The results show that, compared to the original YOLOv8, our YOLOv8-GABNet model reduces the parameter count by 43.6% and increases the mean Average Precision (mAP50) by 4.3%. Moreover, the model size was reduced from 50.1 MB to 30.2 MB, facilitating deployment on mobile devices. When compared with mainstream models like YOLOv5s, Faster R-CNN, SSD, YOLOv9t, and YOLOv10n, the YOLOv8-GABNet model demonstrates superior performance in terms of size and accuracy, offering an optimal balance between performance, size, and speed. This study confirms that the model effectively identifies the common diseases and nutritional deficiencies of citrus from Conghua’s “Citrus Planet”. Future deployment to mobile devices will provide farmers with instant and precise support.

Keywords: citrus; disease recognition; nutritional deficiency recognition; YOLOv8; YOLOv8-GABNet; lightweight (search for similar items in EconPapers)
JEL-codes: Q1 Q10 Q11 Q12 Q13 Q14 Q15 Q16 Q17 Q18 (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2077-0472/14/11/1964/pdf (application/pdf)
https://www.mdpi.com/2077-0472/14/11/1964/ (text/html)

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:gam:jagris:v:14:y:2024:i:11:p:1964-:d:1512381

Access Statistics for this article

Agriculture is currently edited by Ms. Leda Xuan

More articles in Agriculture from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().