关键词:
目标检测
茶叶嫩芽
分级识别
YOLOv7
摘要:
茶叶嫩芽分选鉴定是实现茶叶自动分级的重要环节,为实现对优质茶叶嫩芽的准确识别,提出了一种基于改进目标检测算法YOLOv7的茶叶嫩芽分级识别方法。首先,在主干网络融入CBAM注意力机制,构建CBAM-ELAN模块,在通道和空间注意模块的协同作用下,降低背景部分的权重,增强对嫩芽特征的提取能力;然后,在Neck侧引入感受野增强(RFE)模块和显式视觉中心(EVC)模块,生成新的感受野关系,提升层内特征的调节能力,增强对嫩芽特征的提取能力。改进YOLOv7算法对于单芽等级茶叶的检测精度均值为89.3%,一芽一叶检测精度均值为88.9%,一芽二叶检测精度均值为95.7%,与原始YOLOv7算法相比,检测精度均值分别提高了1.3、1.3和1.5个百分点。该方法在兼顾准确率的前提下,实现了端到端的目标检测和对优质茶苗不同姿态的识别,可为茶叶嫩芽的分级和识别提供理论基础。