一种基于深度学习的茶叶虫害识别方法

    公开(公告)号:CN119206486A

    公开(公告)日:2024-12-27

    申请号:CN202411233613.X

    申请日:2024-09-04

    Abstract: 本发明提供了茶叶虫害识别技术领域的一种基于深度学习的茶叶虫害识别方法,包括:步骤S1、采集大量茶叶图像进行预处理后构建数据集;步骤S2、将数据集划分为训练集、测试集以及验证集;步骤S3、基于YOLOv8n网络的输入模块、特征提取模块、颈部模块以及头部模块创建一茶叶虫害识别模型,设定茶叶虫害识别模型的损失函数;特征提取模块设有SCConv单元构成;颈部模块设有EMA单元;头部模块设有边界框回归层以及分类层;步骤S4、通过训练集对茶叶虫害识别模型进行训练,通过测试集对茶叶虫害识别模型进行测试,通过验证集对茶叶虫害识别模型进行验证;步骤S5、通过茶叶虫害识别模型进行茶叶虫害识别。本发明的优点在于:极大的提升了茶叶虫害识别准确度。

Patent Agency Ranking