湘南学院学报2024,Vol.45Issue(2) :27-33.DOI:10.3969/j.issn.1672-8173.2024.02.005

融合剪枝与知识蒸馏的农业病虫害检测

潘伟 朱镕 史润发 胡国华 连顺 梅腱 何凤莲
湘南学院学报2024,Vol.45Issue(2) :27-33.DOI:10.3969/j.issn.1672-8173.2024.02.005

融合剪枝与知识蒸馏的农业病虫害检测

潘伟 1朱镕 1史润发 1胡国华 1连顺 2梅腱 1何凤莲3
扫码查看

作者信息

  • 1. 合肥大学先进制造工程学院,安徽合肥 230000
  • 2. 讯飞科技股份有限公司,安徽 合肥 230088
  • 3. 安徽医科大学第二附属医院,安徽 合肥 230031
  • 折叠

摘要

自动化监测农产品病虫害情况,可以早期发现问题并采取防控措施,减少农作物的损失.一种基于YOLOv7的检测方法以及一种通用的YOLO模型剪枝流程被用于解决农业病虫害检测算法冗杂、计算量大等问题.具体流程是:对农业数据集进行预训练初始化的YOLOv7模型;应用ResRep剪枝以获得稀疏化的小模型;设计带有中间层知识传递的Neck蒸馏训练策略恢复小模型精度,实现无损剪枝.以包含豆类作物角叶斑点病、草莓花枯病、番茄蜘蛛螨等12类作物病虫害真实图片的数据集为样本,实验结果表明,剪枝后模型的计算量和参数量分别减少了62%和76%,同时平均精度和推理速度分别提升了3.9%和40.4%.研究对提升农业病虫害问题智能检测等新质生产力具有重要的现实参考价值.

关键词

农业病虫害检测/模型剪枝/知识蒸馏/新质生产力/人工智能

引用本文复制引用

基金项目

安徽省高等学校省级质量工程项目(2022sx128)

安徽省高等学校省级质量工程项目(2022jyxm735)

出版年

2024
湘南学院学报
湘南学院

湘南学院学报

影响因子:0.215
ISSN:1672-8173
参考文献量18
段落导航相关论文