English

中国农机化学报

中国农机化学报 ›› 2024, Vol. 45 ›› Issue (12): 140-147.DOI: 10.13733/j.jcam.issn.20955553.2024.12.021

• 农业信息化工程 • 上一篇    下一篇

融合注意力及多重知识迁移的茶叶病害轻量化检测方法

毛致颖1,刘宇航2,杨春勇1, 2, 3,田永胜1,倪文军1, 2, 3,王曦照2   

  1. (1. 中南民族大学电子信息工程学院,武汉市,430074; 2. 智能物联技术湖北省工程研究中心,武汉市,430074; 3. 智能无线通信湖北省重点实验室,武汉市,430074)
  • 出版日期:2024-12-15 发布日期:2024-12-02
  • 基金资助:
    湖北省自然科学基金创新群体项目(2024AFA030)

A lightweight detection method for tea disease by merging attention and multiple-knowledge-transfer

Mao Zhiying1, Liu Yuhang2, Yang Chunyong1, 2, 3, Tian Yongsheng1, Ni Wenjun1, 2, 3, Wang Xizhao2   

  1. (1. College of Electronics and Information Engineering, South-Central Minzu University, Wuhan, 430074, China; 
    2. Hubei Engineering Research Center of Intelligent Internet of Things Technology, Wuhan, 430074, China;
    3. Hubei Key Laboratory of Intelligent Wireless Communications, Wuhan, 430074, China)

  • Online:2024-12-15 Published:2024-12-02

摘要: 茶树病虫害是影响茶叶产量及品质的主要原因,精准检测茶叶病虫害种类是当前国内的热点问题之一。针对传统目标检测网络模型参数量大、精确率低导致工业部署困难的问题,建立茶叶病虫害表型图像数据集;对网络模型进行轻量化处理,优裁基于知识蒸馏的多重知识迁移训练模型;构建基于视觉注意力模块(CSA)的YOLOv5目标检测网络模型,优化茶叶病虫害检测方法。结果表明,添加视觉注意力模块(CSA)的YOLOv5目标检测模型与YOLOv5网络模型、添加传统注意力模块SE、CBAM模块的YOLOv5网络模型相比较,其平均准确率分别提高3.1%,1.1%,1%。对比蒸馏前学生模型,构建的模型最佳准确率提升 4.1%,对比教师模型,模型容量降低5.4 MB,单帧图片推理时间下降 35%。设计的网络模型在不损失准确率的情况下,降低网络计算的开销,可为资源受限的农业信息化领域边缘计算系统提供植入可能。

关键词: 茶叶病虫害, 注意力模块, 知识迁移, 轻量化, 农业信息化边缘计算

Abstract:

Tea plant diseases and pests are the main factors affecting tea production and quality. The accurate detection  of tea diseases and pests is one of the current hot issues in China. Aiming at the problem that the traditional target detection network model is difficult to deploy in industry due to the large number of parameters and low accuracy, the phenotype image dataset of tea diseases and pests is established, the network model is lightened, and the multi-knowledge transfer training model based on knowledge distillation is optimized. A YOLOv5 target detection network model based on the visual attention module (CSA) is constructed, and the tea disease and pest detection method is optimized. The experimental results show that the YOLOv5 target detection model with the visual attention module (CSA) constructed in this paper, compared with the YOLOv5 network model, and the YOLOv5 network model with the traditional attention modules SE and CBAM, respectively, improves the average accuracy by 3.1%, 1.1%, and 1%. Compared with the pre-distilled student model, the model constructed in this paper achieves a maximum accuracy improvement of 4.1%. Compared with the teacher model, the model capacity is reduced by 5.4 MB, and the single-frame image inference time is reduced by 35%. The network model designed in this paper reduces the computational overhead without sacrificing accuracy and can provide an implantation possibility for resource-limited edge computing systems in the field of agricultural informatization.

Key words: tea pests and diseases, attention module, knowledge transfer, lightweight, agricultural information edge computing

中图分类号: