基于YOLOv11的轻量化黄瓜病害检测模型

Lightweight cucumber disease detection model based on YOLOv11

  • 摘要: 针对黄瓜病害检测模型计算量大、难以在资源受限场景部署的问题,提出轻量化模型WDL‑YOLOv11。该模型以YOLOv11n为基础,引入WaveletPool模块以降低计算量并保留细粒度特征,引入DySample上采样增强小目标表征,构建Detect_LSCD检测头,实现参数共享与多尺度特征校准。在包含5类黄瓜病害及健康叶片的5191张图像上进行相关实验。结果表明:WDL‑YOLOv11的mAP@0.5达88.2%,参数量仅有4.1 MB,计算量为4.6 GFLOPs;相比YOLOv11n,参数量和计算量分别减少21.2%和28.1%,精度仅下降0.5个百分点。该模型在参数量和计算量上均优于Rtdetr‑l及YOLO系列轻量模型,且检测精度居前,并展现出良好的架构泛化性与病斑聚焦能力。

     

    Abstract: To address the issues of high computational cost and limited deployability in resource-constrained scenarios for cucumber disease detection, a lightweight model, WDL‑YOLOv11, was proposed. Based on YOLOv11n, a WaveletPool module was introduced to reduce computational cost while preserving fine‑grained features, a DySample upsampling operation was introduced to enhance the representation of small objects, and a Detect_LSCD detection head was constructed to achieve parameter sharing and multi‑scale feature calibration. Experiments were conducted on 5191 images containing five types of cucumber diseases and healthy leaves. The results showed that WDL‑YOLOv11 attained an mAP@0.5 of 88.2%, with only 4.1 MB parameters and 4.6 GFLOPs of computational cost. In comparison with YOLOv11n, the number of parameters and the computational cost were reduced by 21.2% and 28.1%, respectively, while the accuracy was decreased by only 0.5 percentage points. The model was superior to Rtdetr‑l and YOLO series lightweight models in terms of both parameter count and computational cost, and ranked among the top in detection accuracy, and exhibited good architectural generalization and disease spot focusing capability.

     

/

返回文章
返回