基于剪枝和知识蒸馏的YOLOv8轻量化苹果叶片病害检测方法
CSTR:
作者:
作者单位:

河南农业大学信息与管理科学学院,郑州 450046

作者简介:

张帅平, E-mail: zsp2292@stu.henau.edu.cn

通讯作者:

尹飞, E-mail: yin.fei@henau.edu.cn

中图分类号:

TP391.4;S661.1

基金项目:

河南省科技攻关项目(242102521027);河南省科技研发计划联合基金项目(222301420113)


A pruning and knowledge distillation-based YOLOv8 lightweight method for detecting leaf diseases in apple
Author:
Affiliation:

College of Information & Management Sciences, Henan Agricultural University, Zhengzhou 450046, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为解决现有苹果叶片检测方法在准确率、实时性及复杂噪声环境下鲁棒性等方面的问题,基于YOLOv8n提出一种轻量化、实时苹果叶片病害检测模型SMPD-YOLO。首先,将SPPFCSPC(spatial pyramid pooling fast cross stage partial CSP)特征金字塔模块嵌入到主干网络,增强特征融合能力。其次,引入MPD-IoU(minimum point distance-IoU)作为边界框回归损失函数,提高模型的精度和收敛速度。再次,通过基于层自适应幅度剪枝(layer adaptive magnitude-based pruning,LAMP)进一步压缩模型体积、降低浮点运算速度。最后,采用通道级知识蒸馏(channel-wise knowledge distillation,CWD)策略,提升检测性能。实验结果显示,改进后SMPD-YOLO模型在IoU(intersection over union)阈值大于0.5时的平均精度均值和帧率分别达到90.20%和133.3帧/s,模型权重、浮点运算速度分别为5.0 MB、7.3×109 s-1。此外,改进模型在强光、弱光及图像模糊复杂噪声环境下仍能展现出优异的鲁棒性。结果表明,SMPD-YOLO模型兼具高准确性、轻量化和实时性,可在资源受限设备上实现高效叶片病害检测。

    Abstract:

    A lightweight, real-time model of detecting leaf diseases in apple, named as SMPD-YOLO, was proposed based on YOLOv8n to solve the problems in existing methods of detecting leaf diseases in apple in terms of accuracy, real-time performance, and robustness under complex and noisy environments. The spatial pyramid pooling fast cross stage partial CSP (SPPFCSPC) module was embedded into the backbone network to enhance the capability of feature fusion. The minimum point distance-IoU (MPD-IoU) was introduced as the bounding box regression loss function to improve the accuracy and convergence speed of the model. The model volume and floating point operations were further reduced via layer adaptive magnitude-based pruning (LAMP). A channel-wise knowledge distillation (CWD) strategy was used to boost detection performance. The results showed that the improved SMPD-YOLO model had a mean average precision(mAP@0.5) of 90.20% and a frame rate of 133.3 frames per second (FPS),while the weight and FLOPs of model was 5.0 MB and 7.3×109 s-1, respectively. The improved model maintained excellent robustness under complex and noisy environments including strong illumination, weak light, and blurred images. It is indicated that the SMPD-YOLO model combines high accuracy, lightweight design, and real-time performance, enabling the efficient detection of leaf diseases on resource-constrained equipment.

    参考文献
    相似文献
    引证文献
引用本文

张帅平,时雷,郑光,王慧新,尹飞.基于剪枝和知识蒸馏的YOLOv8轻量化苹果叶片病害检测方法[J].华中农业大学学报,2026,45(3):98-114

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2025-09-25
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-06-17
  • 出版日期:
文章二维码