首页 | 本学科首页   官方微博 | 高级检索  
     检索      

基于多头自注意力机制的茶叶采摘点语义分割算法
引用本文:宋彦,杨帅,郑子秋,宁井铭.基于多头自注意力机制的茶叶采摘点语义分割算法[J].农业机械学报,2023,54(9):297-305.
作者姓名:宋彦  杨帅  郑子秋  宁井铭
作者单位:安徽农业大学;茶树生物学与资源利用国家重点实验室
基金项目:国家重点研发计划项目(2021YFD1601102)
摘    要:茶叶采摘点定位是茶叶选择性采摘的关键技术之一,在茶树采摘场景中,存在采摘点尺度小、背景干扰大、光照情况复杂等问题,导致准确分割茶叶采摘点成为难题。本研究针对茶园场景下采摘点精确分割问题,构建了一种基于多头自注意力机制结合多尺度特征融合的语义分割算法——RMHSA-NeXt。首先使用ConvNeXt卷积神经网络提取图像特征;其次构造基于残差和多头自注意力机制的注意力模块,将模型注意力集中于分割目标,增强重要特征的表达;再次通过多尺度结构(Atrous spatial pyramid pooling, ASPP)将不同尺度的特征进行融合,在其中针对采摘点特性,在融合过程中使用条状池化(Strip pooling),减少无用特征的获取;最后通过卷积以及上采样等操作完成信息的解码,得出分割结果。试验表明,茶园环境下该模型可以对采摘点进行有效分割,模型的像素准确率达到75.20%,平均区域重合度为70.78%,运行速度达到8.97 f/s。基于相同测试集将本文模型与HRNet V2、EfficientUNet++、DeeplabV3+、BiSeNet V2模型进行对比,结果表明相比于其他模型同时...

关 键 词:茶叶采摘点  语义分割  注意力机制  多尺度结构
收稿时间:2023/2/24 0:00:00

Semantic Segmentation Algorithm Based Multi-headed Self-attention for Tea Picking Points
SONG Yan,YANG Shuai,ZHENG Ziqiu,NING Jingming.Semantic Segmentation Algorithm Based Multi-headed Self-attention for Tea Picking Points[J].Transactions of the Chinese Society of Agricultural Machinery,2023,54(9):297-305.
Authors:SONG Yan  YANG Shuai  ZHENG Ziqiu  NING Jingming
Institution:Anhui Agricultural University; State Key Laboratory of Tea Plant Biology and Utilization
Abstract:
Keywords:tea picking point  semantic segmentation  attention mechanism  multi-scale structure
点击此处可从《农业机械学报》浏览原始摘要信息
点击此处可从《农业机械学报》下载免费的PDF全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号