CEFW-YOLO: A High-Precision Model for Plant Leaf Disease Detection in Natural Environments

Abstract

The accurate and rapid detection of apple leaf diseases is a critical component of precision management in apple orchards. The existing deep-learning-based detection algorithms for apple leaf diseases typically demand high computational resources, which limits their practical applicability in orchard environments. Furthermore, the detection of apple leaf diseases in natural settings faces significant challenges due to the diversity of disease types, the varied morphology of affected areas, and the influence of factors such as lighting variations, leaf occlusions, and differences in disease severity. To address the above challenges, we constructed an apple leaf disease detection (ALD) dataset, which was collected from real-world scenarios, and we applied data augmentation techniques, resulting in a total of 9808 images. Based on the ALD dataset, we proposed a lightweight YOLO11n-based detection network, named CEFW-YOLO, designed to tackle the current issues in apple leaf disease identification. First, we designed a novel channel-wise squeeze convolution (CWSConv), which employs channel compression and standard convolution to reduce computational resource consumption, enhance the detection of small objects, and improve the model’s adaptability to the morphological diversity of apple leaf diseases and complex backgrounds. Second, we developed an enhanced cross-channel attention (ECCAttention) module and integrated it into the C2PSA_ECCAttention module. By extracting global information, combining horizontal and vertical convolutions, and strengthening cross-channel interactions, this module enables the model to more accurately capture disease features on apple leaves, thereby enhancing detection accuracy and robustness. Additionally, we introduced a new fine-grained multi-level linear attention (FMLAttention) module, which utilizes multi-level asymmetric convolutions and linear attention mechanisms to improve the model’s ability to capture fine-grained features and local details critical for disease detection. Finally, we incorporated the Wise-IoU (WIoU) loss function, which enhances the model’s ability to differentiate overlapping targets across multiple scales. A comprehensive evaluation of CEFW-YOLO was conducted, comparing its performance against state-of-the-art (SOTA) models. CEFW-YOLO achieved a 20.6% reduction in computational complexity. Compared to the original YOLO11n, it improved detection precision by 3.7%, with the [email protected] and [email protected]:0.95 increasing by 7.6% and 5.2%, respectively. Notably, CEFW-YOLO outperformed advanced SOTA algorithms in apple leaf disease detection, underscoring its practical application potential in real-world orchard scenarios.