Highly Accurate and Lightweight Detection Model of Apple Leaf Diseases Based on YOLO

Abstract

To mitigate problems concerning small-sized spots on apple leaves and the difficulties associated with the accurate detection of spot targets exacerbated by the complex backgrounds of orchards, this research used alternaria leaf spots, rust, brown spots, gray spots, and frog eye leaf spots on apple leaves as the research object and proposed the use of a high-accuracy detection model YOLOv5-Res (YOLOv5-Resblock) and lightweight detection model YOLOv5-Res4 (YOLOv5-Resblock-C4). Firstly, a multiscale feature extraction module, ResBlock (residual block), was designed by combining the Inception multi-branch structure and ResNet residual idea. Secondly, a lightweight feature fusion module C4 (CSP Bottleneck with four convolutions) was designed to reduce the number of model parameters while improving the detection ability of small targets. Finally, a parameter-streamlining strategy based on an optimized model architecture was proposed. The experimental results show that the performance of the YOLOv5-Res model and YOLOv5-Res4 model is significantly improved, with the mAP0.5 values increasing by 2.8% and 2.2% compared to the YOLOv5s model and YOLOv5n model, respectively. The sizes of the YOLOv5-Res model and YOLOv5-Res4 model are only 10.8 MB and 2.4 MB, and the model parameter counts are reduced by 22% and 38.3% compared to the YOLOv5s model and YOLOv5n model.

Keywords

• leaf disease detection
• YOLO
• residual block
• parameter streamlining strategy