Journal of Henan Agricultural Sciences ›› 2023, Vol. 52 ›› Issue (5): 162-169.DOI: 10.15933/j.cnki.1004-3268.2023.05.019

• Agricultural Information and Engineering Agricultural Product... • Previous Articles     Next Articles

Research and Application of Lightweight Yolov7‐TSA Network in Tea Disease Detection and Identification

LI Weihao1,ZHAN Wei1,ZHOU Wan1,HAN Tao1,WANG Peiwen1,LIU Hu1,XIONG Mengyuan1,SUN Yong1,2   

  1. (1.School of Computer Science,Yangtze University,Jingzhou 434023,China;2.Jingzhou Yingtuo Technology Co.,Ltd.,Jingzhou 434023,China)
  • Received:2023-01-28 Published:2023-05-15 Online:2023-06-09

轻量型Yolov7-TSA 网络在茶叶病害检测识别中的研究与应用

李伟豪1,詹炜1,周婉1,韩涛1,王佩文1,刘虎1,熊梦园1,孙泳1,2   

  1. (1.长江大学计算机科学学院,湖北 荆州 434023;2.荆州市鹰拓科技有限公司,湖北 荆州 434023)
  • 通讯作者: 詹炜(1977-),男,湖北英山人,教授,博士,主要从事计算机视觉技术应用与植物保护研究。E-mail:zhanwei814@yangtzeu.edu.cn
  • 作者简介:李伟豪(1999-),男,山西临汾人,在读硕士研究生,研究方向:计算机视觉技术应用与植物保护。E-mail:2021710637@yangtzeu.edu.cn
  • 基金资助:
    国家自然科学基金面上项目(62276032);中国高校产学研创新基金——新一代信息技术创新项目(2020ITA03012)

Abstract: Aiming at the problems of low accuracy,slow model running speed and no detection function of the existing tea disease identification methods,a novel Yolov7‐TSA(Yolov7‐Tiny‐SiLU‐Attention)lightweight network architecture was proposed to detect and classify tea diseases.Specifically,the LeakReLU activation function in the Yolov7‐T network was replaced with the SiLU activation function to improve detection accuracy and prevent overfitting.Meanwhile,the feature perception of target contour and spatial location was further improved by fusing the coordinate attention mechanism. Experiments on the dataset containing eight tea diseases(including healthy tea leaves) showed that the recognition accuracy of Yolov7‐TSA network reached 94.2%,which was 3.2 and 1.2 percentage points higher compared to Yolov7‐T and Yolov7 networks,respectively.Furthermore,it showed significant results in terms of parameters,floating point operations,model size and inference time per image,which were reduced by 83%,87%,83% and 34%,respectively,compared to the Yolov7 network. The network model achieves the detection and classification of tea diseases when balancing recognition accuracy and real‐time performance.

Key words: Tea disease, Disease detection, Disease classification, Yolov7‐TSA, Coordinate attention, Lightweight network, Computer vision, Smart agriculture

摘要: 针对现有茶叶病害识别方法准确率低、模型运行速度慢和缺乏检测功能等问题,提出一种新的Yolov7-TSA(Yolov7-Tiny-SiLU-Attention)轻量型网络架构对茶叶病害检测和分类。将Yolov7-T网络中的LeakReLU激活函数替换为SiLU激活函数,以提升检测精度,并防止过拟合。同时,通过融合坐标注意力机制进一步提升对目标轮廓和空间位置的特征感知能力。在含8种茶叶病害(含健康茶叶)的数据集上试验。结果表明,Yolov7-TSA网络对茶叶病害的识别准确率达到了94.2%,与Yolov7-T、Yolov7网络相比,分别提升了3.2、1.2个百分点。另外,Yolov7-TSA网络在参数量、浮点运算数、模型大小和单张图片推理时间方面表现出了显著的效果,与Yolov7网络相比,其分别降低了83%、87%、83%和34%。该网络模型实现了对茶叶病害的检测与分类,同时平衡了识别准确率和实时性能。

关键词: 茶叶病害, 病害检测, 病害分类, Yolov7-TSA, 坐标注意力, 轻量型网络, 计算机视觉, 智慧农业

CLC Number: