Improved YOLOv8-Based Algorithm for Citrus Leaf Disease Detection

In view of the small difference between citrus leaf diseases which can lead to false inspection and missed inspection, an improved YOLOv8 citrus leaf disease detection algorithm is proposed. The proposed approach uses YOLOv8n as the base model and introduces adaptive convolution into the Backbone, a...

Full description

Saved in:
Bibliographic Details
Main Authors: Zhengbing Zheng, Yibang Zhang, Luchao Sun
Format: Article
Language:English
Published: IEEE 2025-01-01
Series:IEEE Access
Subjects:
Online Access:https://ieeexplore.ieee.org/document/11031421/
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1849425980292071424
author Zhengbing Zheng
Yibang Zhang
Luchao Sun
author_facet Zhengbing Zheng
Yibang Zhang
Luchao Sun
author_sort Zhengbing Zheng
collection DOAJ
description In view of the small difference between citrus leaf diseases which can lead to false inspection and missed inspection, an improved YOLOv8 citrus leaf disease detection algorithm is proposed. The proposed approach uses YOLOv8n as the base model and introduces adaptive convolution into the Backbone, allowing the model to dynamically prioritize different disease features. This modification increases the model’s parameter count without adding to the computational cost in terms of floating-point operations. Next, a fast convolution layer is implemented to replace the original C2f module, improving both detection accuracy and computational efficiency. In addition, a multi-scale fusion attention module is incorporated into the Neck section, which effectively integrates disease-related information from different receptive fields, thereby boosting the model’s ability to detect a wider variety of diseases. To further enhance adaptability, the traditional Intersection over Union (IoU) is replaced with the Wise-IoU loss function, improving the model’s performance across different disease types. The algorithm model uses a self-built data set containing 1,810 images, with 481 images of Huanglongbing, 430 images of magnesium deficiency, 451 images of anthracnose, and 448 images of black spot. In order to maintain the randomness of the experimental data, this experiment divided the data set into training set, validation set and test set with a ratio of 7:2:1, which contained 1267 images, 362 images and 181 images respectively. Experimental results show that the improved algorithm achieves an accuracy of 93.2% on the test set, representing a 1.2 percentage increase over the original model, while also reducing the parameter volume and computational complexity by 7% and 7.4% respectively. The improved model notably decreases the missed detection rate under occlusion and improves the detection of small targets. Furthermore, it strikes an effective balance between enhancing detection accuracy, reducing model complexity, and maintaining a lightweight architecture, making them well-suited for efficient citrus leaf disease detection.
format Article
id doaj-art-53d73a11c03f444ea58f7c0df6affc65
institution Kabale University
issn 2169-3536
language English
publishDate 2025-01-01
publisher IEEE
record_format Article
series IEEE Access
spelling doaj-art-53d73a11c03f444ea58f7c0df6affc652025-08-20T03:29:35ZengIEEEIEEE Access2169-35362025-01-011310588810590010.1109/ACCESS.2025.357907411031421Improved YOLOv8-Based Algorithm for Citrus Leaf Disease DetectionZhengbing Zheng0https://orcid.org/0000-0003-4669-3511Yibang Zhang1https://orcid.org/0009-0002-6976-8416Luchao Sun2https://orcid.org/0009-0005-5241-3688School of Physics and Telecommunication Engineering, Shaanxi University of Technology, Hanzhong, Shaanxi, ChinaSchool of Physics and Telecommunication Engineering, Shaanxi University of Technology, Hanzhong, Shaanxi, ChinaSchool of Physics and Telecommunication Engineering, Shaanxi University of Technology, Hanzhong, Shaanxi, ChinaIn view of the small difference between citrus leaf diseases which can lead to false inspection and missed inspection, an improved YOLOv8 citrus leaf disease detection algorithm is proposed. The proposed approach uses YOLOv8n as the base model and introduces adaptive convolution into the Backbone, allowing the model to dynamically prioritize different disease features. This modification increases the model’s parameter count without adding to the computational cost in terms of floating-point operations. Next, a fast convolution layer is implemented to replace the original C2f module, improving both detection accuracy and computational efficiency. In addition, a multi-scale fusion attention module is incorporated into the Neck section, which effectively integrates disease-related information from different receptive fields, thereby boosting the model’s ability to detect a wider variety of diseases. To further enhance adaptability, the traditional Intersection over Union (IoU) is replaced with the Wise-IoU loss function, improving the model’s performance across different disease types. The algorithm model uses a self-built data set containing 1,810 images, with 481 images of Huanglongbing, 430 images of magnesium deficiency, 451 images of anthracnose, and 448 images of black spot. In order to maintain the randomness of the experimental data, this experiment divided the data set into training set, validation set and test set with a ratio of 7:2:1, which contained 1267 images, 362 images and 181 images respectively. Experimental results show that the improved algorithm achieves an accuracy of 93.2% on the test set, representing a 1.2 percentage increase over the original model, while also reducing the parameter volume and computational complexity by 7% and 7.4% respectively. The improved model notably decreases the missed detection rate under occlusion and improves the detection of small targets. Furthermore, it strikes an effective balance between enhancing detection accuracy, reducing model complexity, and maintaining a lightweight architecture, making them well-suited for efficient citrus leaf disease detection.https://ieeexplore.ieee.org/document/11031421/Wise-IoU loss functionfeature fusionadaptive convolutiondeep learningmulti-scale fusion attention module
spellingShingle Zhengbing Zheng
Yibang Zhang
Luchao Sun
Improved YOLOv8-Based Algorithm for Citrus Leaf Disease Detection
IEEE Access
Wise-IoU loss function
feature fusion
adaptive convolution
deep learning
multi-scale fusion attention module
title Improved YOLOv8-Based Algorithm for Citrus Leaf Disease Detection
title_full Improved YOLOv8-Based Algorithm for Citrus Leaf Disease Detection
title_fullStr Improved YOLOv8-Based Algorithm for Citrus Leaf Disease Detection
title_full_unstemmed Improved YOLOv8-Based Algorithm for Citrus Leaf Disease Detection
title_short Improved YOLOv8-Based Algorithm for Citrus Leaf Disease Detection
title_sort improved yolov8 based algorithm for citrus leaf disease detection
topic Wise-IoU loss function
feature fusion
adaptive convolution
deep learning
multi-scale fusion attention module
url https://ieeexplore.ieee.org/document/11031421/
work_keys_str_mv AT zhengbingzheng improvedyolov8basedalgorithmforcitrusleafdiseasedetection
AT yibangzhang improvedyolov8basedalgorithmforcitrusleafdiseasedetection
AT luchaosun improvedyolov8basedalgorithmforcitrusleafdiseasedetection