基于改进YOLOv8n的茶树嫩芽识别

文献类型： 中文期刊

第一作者： 杨肖委

作者： 杨肖委;沈强;罗金龙;张拓;杨婷;戴宇樵;刘忠英;李琴;王家伦

作者机构：

关键词： 深度学习;茶树嫩芽;图像识别;YOLOv8n;注意力机制;采摘机器人

期刊名称： 茶叶科学

ISSN： 1000-369X

年卷期： 2024 年 44 卷 006 期

页码： 949-959

收录情况： 北大核心 ; CSCD

摘要： 在复杂自然环境下对茶树嫩芽进行精确识别是实现农业机器人智能化采摘茶树嫩芽的关键技术之一。针对茶园复杂环境下茶树嫩芽识别率低的问题，提出了一种基于改进YOLOv8n的茶树嫩芽检测方法。使用荣耀80手机采集茶树嫩芽图片，并对图片进行标注，按照训练集、验证集、测试集8∶1∶1的比例划分数据集。为有效提取嫩芽特征并减少模型冗余计算和内存访问，采用FasterNet模型替换YOLOv8n网络结构的骨干网络进行特征提取；为抑制茶园环境背景信息、增强模型对嫩芽的特征提取能力，在主干网络尾部（SPPF模块后）引入全局注意力机制（Global attention mechanism,GAM）；在Neck网络中引入上下文引导（Context guided,CG）模块，学习茶树嫩芽局部特征和周围环境的联合特征，进一步提高茶树嫩芽的识别准确率。利用构建的茶树嫩芽数据集对改进的YOLOv8n算法进行训练和测试。消融试验验证结果显示，FasterNet网络、GAM注意力机制和CG模块均有效提高了YOLOv8n模型的识别准确率。改进的YOLOv8n模型在多类别的茶树嫩芽数据集上平均准确率（Mean average accuracy,mAP）为94.3%；相较于原YOLOv8n模型，对茶树单芽、一芽一叶、一芽二叶的识别mAP分别提高了2.2个百分点、1.6个百分点、2.7个百分点。以YOLOv3-tiny、YOLOv3、YOLOv5m、YOLOv7-tiny、YOLOv7和YOLOv8n模型进行对比试验，结果显示，改进的YOLOv8n模型对茶树嫩芽识别的效果最佳，说明改进的YOLOv8n模型能有效提升茶树嫩芽识别准确率。研究结果可为智能化茶叶采摘机器人的开发提供技术支撑。

分类号： S225.99

  • 相关文献

[1]基于改进YOLOv5l的设施番茄3D信息检测方法. 林森,许童羽,葛禹豪,马璟,孙添龙,赵春江. 2024

[2]基于注意力机制及多尺度特征融合的番茄叶片缺素图像分类方法. 韩旭,赵春江,吴华瑞,朱华吉,张燕. 2021

[3]基于改进YOLOV5s网络的奶牛多尺度行为识别方法. 白强,高荣华,赵春江,李奇峰,王荣,李书琴. 2022

[4]基于YOLOX改进模型的茶叶嫩芽识别方法. 俞龙,黄楚斌,唐劲驰,黄浩宜,周运峰,黄永权,孙佳琪. 2022

[5]基于注意力机制的农业文本命名实体识别. 赵鹏飞,赵春江,吴华瑞,王维. 2021

[6]基于改进YOLO v5复杂场景下肉鹅姿态的检测算法研究. 刘璎瑛,曹晅,郭彬彬,陈慧杰,戴子淳,龚长万. 2023

[7]基于YOLOv5s-SE和通道剪枝的虫咬紫金蝉茶检测方法研究. 戴佳兵,宋春芳,凌彩金,李臻锋,孙崇高. 2024

[8]基于改进DeepLabV3+的梨树冠层分割方法. 陈鲁威,曾锦,袁全春,夏烨,潘健,吕晓兰. 2024

[9]基于坐标注意力机制与高效边界框回归损失的线虫快速识别. 陆健强,梁效,余超然,兰玉彬,邱洪斌,黄捷伟,尹梓濠,陈慧洁,郑胜杰. 2022

[10]基于深度学习分类模型的4种果树物候期识别. 钟丹,李宗南,王思,黄平,邱霞,蒋怡. 2023

[11]基于TBTA网络的高光谱图像分类. 唐婷,潘新,罗小玲,闫伟红. 2023

[12]基于高光谱成像和Att-BiGRU-RNN的柑橘病叶分类. 吴叶兰,管慧宁,廉小亲,于重重,廖禺. 2023

[13]基于改进YOLOv5s的复杂环境下新梅检测方法. 董耿耿,陈小康,樊湘鹏,周建平,姜宏,崔超. 2024

[14]基于卷积神经网络的农机图像自动识别研究. 雷雪梅,张光强,姚旗,刘伟渭,邱帅. 2022

[15]基于迁移学习和金字塔卷积网络的河蟹个体图像识别方法研究. 冯裕清,杨信廷,徐大明,罗娜,陈枫,孙传恒. 2022

[16]基于YOLOv3深度卷积神经网络的田间百香果定位. 林营志,卢依琳,刘现. 2019

[17]基于性诱和深度学习的草地贪夜蛾成虫自动识别计数方法. 邱荣洲,赵健,何玉仙,陈韶萍,黄美玲,池美香,梁勇,翁启勇. 2021

[18]基于Faster R-CNN的美国白蛾图像识别模型研究. 薛大暄,张瑞瑞,陈立平,陈梅香,徐刚. 2020

[19]卷积神经网络在农业病虫害识别中的应用. 张耀丽,许宁,宋裕民,孟庆山,侯旭,李虎. 2023

[20]采用组合增强的YOLOX-ViT协同识别温室内番茄花果. 吕志远,张付杰,魏晓明,黄媛,李晶晶,张钟莉莉. 2023

作者其他论文 更多>>

微信小程序